Skip to content

Commit 0910998

Browse files
authored
Disable rapidsai until compatible with tf cudatoolkit
1 parent 2ec8ae9 commit 0910998

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

Dockerfile.tmpl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,10 +89,11 @@ RUN conda config --add channels nvidia && \
8989
{{ if eq .Accelerator "gpu" }}
9090

9191
# 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 }}
92+
# b/267180053: RapidsAI (cudf/cuml) are not compatible with the latest tensorflow cudatoolkit version.
93+
# RUN pip uninstall -y pyarrow && \
94+
# mamba install -y cudf cuml && \
95+
# /tmp/clean-layer.sh
96+
# {{ end }}
9697

9798
# Install implicit
9899
{{ if eq .Accelerator "gpu" }}

0 commit comments

Comments
 (0)