Spaces:

jbilcke-hf
/

train-robots-with-mujoco

Paused

jbilcke-hf commited on Jul 1

Commit

5c89d33

verified ·

1 Parent(s): ced4d32

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -68,8 +68,11 @@ RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py31
  && rm ~/miniconda.sh \
  && 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

  && rm ~/miniconda.sh \
  && conda clean -ya
+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