Drop CUDA to 12.1

rainx0r · Aug 19, 2024 · 1d50f6b · 1d50f6b
1 parent 4bc2dd6
commit 1d50f6b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/metaworld-jax/requirements.txt b/metaworld-jax/requirements.txt
@@ -1,10 +1,10 @@
 # Cuda
-nvidia-cublas-cu12~=12.2.0
-nvidia-cuda-cupti-cu12~=12.2.0
-nvidia-cuda-nvcc-cu12~=12.2.0
-nvidia-cuda-runtime-cu12~=12.2.0
-nvidia-cusparse-cu12~=12.2.0
-nvidia-nvjitlink-cu12~=12.2.0
+nvidia-cublas-cu12~=12.1.0
+nvidia-cuda-cupti-cu12~=12.1.0
+nvidia-cuda-nvcc-cu12~=12.1.0
+nvidia-cuda-runtime-cu12~=12.1.0
+nvidia-cusparse-cu12~=12.1.0
+nvidia-nvjitlink-cu12~=12.1.0
 
 # Jax
 jax[cuda12]==0.4.31