We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 962daca commit 6732bc3Copy full SHA for 6732bc3
Dockerfile.tmpl
@@ -91,8 +91,6 @@ RUN conda config --add channels nvidia && \
91
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
92
RUN pip uninstall -y pyarrow && \
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 && \
96
/tmp/clean-layer.sh
97
{{ end }}
98
0 commit comments