Spaces:

jbilcke-hf
/

train-robots-with-mujoco

Runtime error

jbilcke-hf HF Staff commited on 13 days ago

Commit

ced4d32

verified ·

1 Parent(s): 5f20d7a

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

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

  && conda clean -ya
 # Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
+RUN conda install nvidia::libcusparse "jaxlib=*=*cuda*" jax -c conda-forge
 WORKDIR $HOME/app