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 87d59c9 commit abaa7f0Copy full SHA for abaa7f0
Dockerfile.tmpl
@@ -90,7 +90,9 @@ RUN conda config --add channels nvidia && \
90
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
91
RUN pip uninstall -y pyarrow && \
92
conda install -c conda-forge mamba && \
93
- mamba install -y cudf cuml cudatoolkit==11.2.2 && \
+ 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