From 1ebc761dfee90346433b30ce5e50ebd039d3b575 Mon Sep 17 00:00:00 2001 From: Siddharth Venkatesan Date: Tue, 19 Nov 2024 09:43:58 -0800 Subject: [PATCH] [pytorch] Update supported versions of pytorch for 0.32.0 --- .github/workflows/nightly_publish.yml | 5 +++-- engines/pytorch/pytorch-engine/README.md | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/nightly_publish.yml b/.github/workflows/nightly_publish.yml index 3f78edde2d0..0478e2fb7d5 100644 --- a/.github/workflows/nightly_publish.yml +++ b/.github/workflows/nightly_publish.yml @@ -175,11 +175,12 @@ jobs: ${{ runner.os }}-gradle- - name: Publish to snapshot repository if: ${{ github.event.inputs.mode == '' || github.event.inputs.mode == 'snapshot' }} + # PT 1.13.1 is for PT 1.x support, and CUDA 11.x support + # PT 2.3.1 is for CUDA 12.1 support + # PT 2.5.1 is for CUDA 12.4 support run: | ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=1.13.1 -Psnapshot - ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.1.2 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.3.1 -Psnapshot - ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.4.0 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.5.1 -Psnapshot ./gradlew clean engines:ml:xgboost:publish -Pgpu -Psnapshot ./gradlew clean publish -Psnapshot diff --git a/engines/pytorch/pytorch-engine/README.md b/engines/pytorch/pytorch-engine/README.md index a2f3ad76283..23e42d41fa3 100644 --- a/engines/pytorch/pytorch-engine/README.md +++ b/engines/pytorch/pytorch-engine/README.md @@ -51,6 +51,7 @@ The following table illustrates which pytorch version that DJL supports: | PyTorch engine version | PyTorch native library version | |------------------------|-------------------------------------------| +| pytorch-engine:0.32.0 | 1.13.1, 2.3.1, **2.5.1** | | pytorch-engine:0.31.0 | 1.13.1, 2.1.2, 2.3.1, 2.4.0, **2.5.1** | | pytorch-engine:0.30.0 | 1.13.1, 2.1.2, 2.3.1, **2.4.0** | | pytorch-engine:0.29.0 | 1.13.1, 2.1.2, 2.2.2, **2.3.1** |