Skip to content

Commit 74046c2

Browse files
authored
Bump cudf/cuml to 21.12
This is the newest version that supports both CUDA 11.X & Python 3.7
1 parent fb69472 commit 74046c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile.tmpl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ RUN conda config --add channels nvidia && \
8989

9090
# b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
9191
RUN pip uninstall -y pyarrow && \
92-
conda install cudf=21.10 cuml=21.10 cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
92+
conda install cudf=21.12 cuml=21.12 cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
9393
/tmp/clean-layer.sh
9494
{{ end }}
9595

0 commit comments

Comments
 (0)