Spaces:

jbilcke-hf
/

train-robots-with-mujoco

Runtime error

jbilcke-hf HF Staff commited on 13 days ago

Commit

d362bd4

verified ·

1 Parent(s): 22c3387

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -71,7 +71,7 @@ RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py31
 ENV CONDA_OVERRIDE_CUDA="12.8"
 # Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
-RUN conda install nvidia::cuda-toolkit nvidia::libcusparse
 RUN conda install "jaxlib=*=*cuda*" jax -c conda-forge
 WORKDIR $HOME/app

 ENV CONDA_OVERRIDE_CUDA="12.8"
 # 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