add cuda 12.1, pytorch 2.1 releases

teddykoker · Dec 29, 2023 · c5474f2 · c5474f2
1 parent 1b672e0
commit c5474f2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/.github/workflows/cuda/install.sh b/.github/workflows/cuda/install.sh
@@ -13,6 +13,9 @@ case "${cuda_version}" in
   "cu118")
     url=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
     ;;
+  "cu121")
+    url=https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda-repo-ubuntu2004-12-1-local_12.1.1-530.30.02-1_amd64.deb
+    ;;
   *)
     >&2 echo "Unsupported cuda_version: ${cuda_version}"
     exit 1

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -20,11 +20,17 @@ jobs:
       matrix:
         os: [ubuntu-20.04]
         python-version: ['3.10', '3.11']
-        torch-version: ['1.13.1', '2.0.1']
-        cuda-version: ['cpu', 'cu117', 'cu118']
+        torch-version: ['1.13.1', '2.0.1', '2.1.0']
+        cuda-version: ['cpu', 'cu117', 'cu118', 'cu121']
         exclude:
           - torch-version: 1.13.1
             cuda-version: cu118
+          - torch-version: 1.13.1
+            cuda-version: cu121
+          - torch-version: 2.0.1
+            cuda-version: cu121
+          - torch-version: 2.1.0
+            cuda-version: cu117
     steps:
       - name: Free disk space
         run: |