One more chance Fix torch CI GPU issue

rapidsai · Oct 9, 2024 · 47a39bc · 47a39bc
1 parent a88e004
commit 47a39bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/test_gpu.sh b/ci/test_gpu.sh
@@ -18,7 +18,7 @@ fi
 echo "Installing pytorch,transformers and pytest to the environment for crossfit tests..."
 mamba install \
   cuda-version=$CUDA_VERSION \
-  "pytorch>=2.0,<=*cuda*"
+  "pytorch>=2.*=*cuda*"
   transformers \
   pytest \
   sentence-transformers  \