diff --git a/engines/pytorch/pytorch-engine/README.md b/engines/pytorch/pytorch-engine/README.md index 4b63eda8ecc..9c5512be8fc 100644 --- a/engines/pytorch/pytorch-engine/README.md +++ b/engines/pytorch/pytorch-engine/README.md @@ -133,8 +133,8 @@ export PYTORCH_FLAVOR=cpu For macOS M1, you can use the following library: -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu:2.3.1:osx-aarch64 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cpu:2.4.0:osx-aarch64 ```xml @@ -142,13 +142,13 @@ For macOS M1, you can use the following library: ai.djl.pytorch pytorch-native-cpu osx-aarch64 - 2.3.1 + 2.4.0 runtime ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` @@ -160,30 +160,30 @@ installed on your GPU machine, you can use one of the following library: #### Linux GPU -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cu121:2.3.1:linux-x86_64 - CUDA 12.1 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cu124:2.4.0:linux-x86_64 - CUDA 12.4 ```xml ai.djl.pytorch - pytorch-native-cu121 + pytorch-native-cu124 linux-x86_64 - 2.3.1 + 2.4.0 runtime ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` ### Linux CPU -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu:2.3.1:linux-x86_64 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cpu:2.4.0:linux-x86_64 ```xml @@ -192,20 +192,20 @@ installed on your GPU machine, you can use one of the following library: pytorch-native-cpu linux-x86_64 runtime - 2.3.1 + 2.4.0 ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` ### For aarch64 build -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu-precxx11:2.3.1:linux-aarch64 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cpu-precxx11:2.4.0:linux-aarch64 ```xml @@ -214,12 +214,12 @@ installed on your GPU machine, you can use one of the following library: pytorch-native-cpu-precxx11 linux-aarch64 runtime - 2.3.1 + 2.4.0 ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` @@ -232,15 +232,15 @@ We also provide packages for the system like Amazonliunx 2 with GLIBC >= 2.17. All the package were built with GCC 7, we provided a newer `libstdc++.so.6.24` in the package that contains `CXXABI_1.3.9` to use the package successfully. -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu-precxx11:2.3.1:linux-x86_64 - CPU +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cpu-precxx11:2.4.0:linux-x86_64 - CPU ```xml ai.djl.pytorch pytorch-jni - 2.3.1-0.29.0 + 2.4.0-0.30.0 runtime ``` @@ -251,13 +251,13 @@ contains `CXXABI_1.3.9` to use the package successfully. ai.djl.pytorch pytorch-native-cpu-precxx11 linux-x86_64 - 2.3.1 + 2.4.0 runtime ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` @@ -273,30 +273,30 @@ For the Windows platform, you can choose between CPU and GPU. #### Windows GPU -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cu121:2.3.1:win-x86_64 - CUDA 12.1 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cu124:2.4.0:win-x86_64 - CUDA 12.4 ```xml ai.djl.pytorch - pytorch-native-cu121 + pytorch-native-cu124 win-x86_64 - 2.3.1 + 2.4.0 runtime ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ``` ### Windows CPU -- ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu:2.3.1:win-x86_64 +- ai.djl.pytorch:pytorch-jni:2.4.0-0.30.0 +- ai.djl.pytorch:pytorch-native-cpu:2.4.0:win-x86_64 ```xml @@ -305,12 +305,12 @@ For the Windows platform, you can choose between CPU and GPU. pytorch-native-cpu win-x86_64 runtime - 2.3.1 + 2.4.0 ai.djl.pytorch pytorch-jni -2.3.1-0.29.0 +2.4.0-0.30.0 runtime ```