Update wheels-generator.yml

OutofAi · Aug 29, 2024 · eca5afb · eca5afb
1 parent 6125a7f
commit eca5afb
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/.github/workflows/wheels-generator.yml b/.github/workflows/wheels-generator.yml
@@ -2,7 +2,7 @@ name: CI
 on:
   workflow_dispatch:
 env:
-  TINY_CUDA_NN_VERSION: "1.7.1"
+  TINY_CUDA_NN_VERSION: "1.7.2"
 jobs:
 
   build-tag:
@@ -37,25 +37,25 @@ jobs:
           - os: windows-2019
             visual_studio: "Visual Studio 16 2019"
             python-version: '3.10'
-            torch-version: '2.3.1'
+            torch-version: '2.4.0'
             cuda-version: "12.1"
             arch: 89
           - os: windows-2019
             visual_studio: "Visual Studio 16 2019"
             python-version: '3.10'
-            torch-version: '2.3.1'
+            torch-version: '2.4.0'
             cuda-version: "12.1"
             arch: 75
           - os: windows-2019
             visual_studio: "Visual Studio 16 2019"
             python-version: '3.10'
-            torch-version: '2.3.1'
+            torch-version: '2.4.0'
             cuda-version: "12.1"
             arch: 70
           - os: windows-2019
             visual_studio: "Visual Studio 16 2019"
             python-version: '3.10'
-            torch-version: '2.3.1'
+            torch-version: '2.4.0'
             cuda-version: "12.1"
             arch: 61
 
@@ -162,17 +162,17 @@ jobs:
           - os: ubuntu-22.04
             cuda: nvidia/cuda:12.1.0-devel-ubuntu22.04
             cudaversion: 121
-            torchversion: 231
+            torchversion: 240
             arch: 75 # for T4 on google colab
           - os: ubuntu-22.04
             cuda: nvidia/cuda:12.1.0-devel-ubuntu22.04
             cudaversion: 121
-            torchversion: 231
+            torchversion: 240
             arch: 70 # for v100 on google colab
           - os: ubuntu-22.04
             cuda: nvidia/cuda:12.1.0-devel-ubuntu22.04
             cudaversion: 121
-            torchversion: 231
+            torchversion: 240
             arch: 89 # for a100 on google colab
     env:
       build_dir: "build"
@@ -203,7 +203,7 @@ jobs:
           pip install setuptools wheel
       - name: Install Pytorch
         run: |
-          pip3 install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
+          pip3 install torch==2.4.0 --index-url https://download.pytorch.org/whl/cu121
       - name: Generate Wheels
         run: |
           cd bindings/torch