diff --git a/metaworld-jax/requirements.txt b/metaworld-jax/requirements.txt
index 42f1c69..95cd0f4 100644
--- 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