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 2ec8ae9 commit 0910998Copy full SHA for 0910998
Dockerfile.tmpl
@@ -89,10 +89,11 @@ RUN conda config --add channels nvidia && \
89
{{ if eq .Accelerator "gpu" }}
90
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
- /tmp/clean-layer.sh
95
-{{ end }}
+# b/267180053: RapidsAI (cudf/cuml) are not compatible with the latest tensorflow cudatoolkit version.
+# RUN pip uninstall -y pyarrow && \
+# mamba install -y cudf cuml && \
+# /tmp/clean-layer.sh
96
+# {{ end }}
97
98
# Install implicit
99
0 commit comments