switch from 118 to 121

pytorch · Oct 4, 2023 · a4231c9 · a4231c9
1 parent d8be0b2
commit a4231c9
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/.github/workflows/release_build.yml b/.github/workflows/release_build.yml
@@ -25,19 +25,19 @@ jobs:
          - os: linux.2xlarge
            python-version: 3.8
            python-tag: "py38"
-           cuda-tag: "cu118"
+           cuda-tag: "cu121"
          - os: linux.2xlarge
            python-version: 3.9
            python-tag: "py39"
-           cuda-tag: "cu118"
+           cuda-tag: "cu121"
          - os: linux.2xlarge
            python-version: '3.10'
            python-tag: "py310"
-           cuda-tag: "cu118"
+           cuda-tag: "cu121"
          - os: linux.2xlarge
            python-version: '3.11'
            python-tag: "py311"
-           cuda-tag: "cu118"
+           cuda-tag: "cu121"
     steps:
     # Checkout the repository to the GitHub Actions runner
     - name: Check ldd --version
@@ -73,11 +73,11 @@ jobs:
     - name: Install PyTorch and CUDA
       shell: bash
       run: |
-        conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu118
+        conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu121
     - name: Install fbgemm
       shell: bash
       run: |
-        conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu118
+        conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu121
     - name: Install Dependencies
       shell: bash
       run: |
@@ -111,7 +111,7 @@ jobs:
       matrix:
         os: [linux.4xlarge.nvidia.gpu]
         python-version: [3.8, 3.9, "3.10", "3.11"]
-        cuda-tag: ["cu118"]
+        cuda-tag: ["cu121"]
     needs: build_on_cpu
     # the glibc version should match the version of the one we used to build the binary
     # for this case, it's 2.26
@@ -176,12 +176,12 @@ jobs:
     - name: Install PyTorch and CUDA
       shell: bash
       run: |
-        conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu118
+        conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu121
     # download wheel from GHA
     - name: Install fbgemm
       shell: bash
       run: |
-        conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu118
+        conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu121
     - name: Install torchmetrics
       shell: bash
       run: |