upgrade torch

linkedin · Dec 6, 2024 · 28f67c5 · 28f67c5
1 parent 163e89b
commit 28f67c5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/amd-ci.yml b/.github/workflows/amd-ci.yml
@@ -58,11 +58,18 @@ jobs:
 
     - name: Setup Dependencies
       run: |
+        python3 -m pip uninstall -y torch torchvision 
+        python3 -m pip install --pre \
+                torch==2.6.0.dev20241113+rocm6.2 \
+                'setuptools-scm>=8' \
+                torchvision==0.20.0.dev20241113+rocm6.2 \
+                --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2
         python3 -m pip install triton==3.1.0 transformers==4.46.3
         python3 -m pip install -e .[dev]
     
     - name: Run Unit Tests
       run: |
+        python3 -m pip list | grep torch
         make test
         make test-convergence