still resolving cuda pytorch incompatibility try 2

StanfordVL · Nov 15, 2024 · c82e59e · c82e59e
1 parent dbd9638
commit c82e59e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -51,7 +51,7 @@ jobs:
           path: omnigibson-src
 
       - name: Install CUDA and PyTorch
-        run: micromamba install cuda=12.4 pytorch
+        run: micromamba install cuda=12.6 pytorch=2.5.1=cuda126_py310he4c8055_303
 
       - name: Install
         working-directory: omnigibson-src