Skip to content

Commit 6732bc3

Browse files
authored
Undo force cudatoolkit, causes inconsistent env
1 parent 962daca commit 6732bc3

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

Dockerfile.tmpl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,6 @@ RUN conda config --add channels nvidia && \
9191
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
9292
RUN pip uninstall -y pyarrow && \
9393
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 && \
9694
/tmp/clean-layer.sh
9795
{{ end }}
9896

0 commit comments

Comments
 (0)