Julian Bilcke
commited on
Commit
·
891d381
1
Parent(s):
4c4343e
something broke automatically, let's fix it manually
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -118,6 +118,11 @@ RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py31
|
|
| 118 |
|
| 119 |
ENV CONDA_OVERRIDE_CUDA="12.8"
|
| 120 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 121 |
# Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
|
| 122 |
RUN conda install nvidia/label/cuda-12.8.1::cuda-toolkit nvidia/label/cuda-12.8.1::libcusparse
|
| 123 |
RUN conda install "jaxlib=*=*cuda*" jax -c conda-forge
|
|
|
|
| 118 |
|
| 119 |
ENV CONDA_OVERRIDE_CUDA="12.8"
|
| 120 |
|
| 121 |
+
# Accept Conda Terms of Service
|
| 122 |
+
RUN conda config --set channel_priority strict \
|
| 123 |
+
&& conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main \
|
| 124 |
+
&& conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
|
| 125 |
+
|
| 126 |
# Make sure that Jax and cuSPARSE are properly installed for CUDA 12.x
|
| 127 |
RUN conda install nvidia/label/cuda-12.8.1::cuda-toolkit nvidia/label/cuda-12.8.1::libcusparse
|
| 128 |
RUN conda install "jaxlib=*=*cuda*" jax -c conda-forge
|