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  \