Add PyTorch v2.3.1

mjun0812 · Jul 17, 2024 · c376ef7 · c376ef7
1 parent bb1a588
commit c376ef7
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -46,6 +46,24 @@ jobs:
             python: "3.10"
             torch_install_cmd: pip install torch==2.2.1 --index-url https://download.pytorch.org/whl/cu121
             path: torch_cpp-1.0.0+torch221cu121-cp310-cp310-linux_x86_64.whl
+
+          # torch 2.3.1
+          - cuda: "11.8.0"
+            python: "3.11"
+            torch_install_cmd: pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu118
+            path: torch_cpp-1.0.0+torch231cu118-cp311-cp311-linux_x86_64.whl
+          - cuda: "11.8.0"
+            python: "3.10"
+            torch_install_cmd: pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu118
+            path: torch_cpp-1.0.0+torch231cu118-cp310-cp310-linux_x86_64.whl
+          - cuda: "12.1.1"
+            python: "3.11"
+            torch_install_cmd: pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
+            path: torch_cpp-1.0.0+torch231cu121-cp311-cp311-linux_x86_64.whl
+          - cuda: "12.1.1"
+            python: "3.10"
+            torch_install_cmd: pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
+            path: torch_cpp-1.0.0+torch231cu121-cp310-cp310-linux_x86_64.whl
 
     steps:
       - name: Maximize build space