Skip to content

Commit abaa7f0

Browse files
authored
force cudatoolkit 11.2.2
1 parent 87d59c9 commit abaa7f0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Dockerfile.tmpl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,9 @@ RUN conda config --add channels nvidia && \
9090
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
9191
RUN pip uninstall -y pyarrow && \
9292
conda install -c conda-forge mamba && \
93-
mamba install -y cudf cuml cudatoolkit==11.2.2 && \
93+
mamba install -y cudf cuml && \
94+
# b/255757999: Force switch the cudatoolkit version back to the existing installation so it is compatible with pytorch
95+
conda remove -y --force cudatoolkit=11.2.72 && mamba install --channel https://conda.anaconda.org/conda-forge/linux-64 cudatoolkit=11.2.2 && \
9496
/tmp/clean-layer.sh
9597
{{ end }}
9698

0 commit comments

Comments
 (0)