From 47a39bc059996c9535ac35fed4d9832331e34b2d Mon Sep 17 00:00:00 2001 From: Vibhu Jawa Date: Tue, 8 Oct 2024 20:56:42 -0700 Subject: [PATCH] One more chance Fix torch CI GPU issue --- ci/test_gpu.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ci/test_gpu.sh b/ci/test_gpu.sh index 1b8f2da..9da821b 100755 --- 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 \