Spaces:

jbilcke-hf
/

train-robots-with-mujoco

Paused

jbilcke-hf commited on Sep 7

Commit

f291b72

verified ·

1 Parent(s): 891d381

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -124,8 +124,9 @@ RUN conda config --set channel_priority strict \
  && conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
 # Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
-RUN conda install nvidia/label/cuda-12.8.1::cuda-toolkit nvidia/label/cuda-12.8.1::libcusparse
-RUN conda install "jaxlib=*=*cuda*" jax -c conda-forge
 WORKDIR $HOME/app

  && conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
 # Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
+# this suddenly stopped to work after 2 months for some reason.. that's a bit frustrating.
+#RUN conda install nvidia/label/cuda-12.8.1::cuda-toolkit nvidia/label/cuda-12.8.1::libcusparse
+#RUN conda install "jaxlib=*=*cuda*" jax -c conda-forge
 WORKDIR $HOME/app