From 98d43263d21c4ed28b274dc6a8f86809bb431070 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Wed, 25 Aug 2021 20:41:09 +0000 Subject: [PATCH] Upgrade cudf/cuml libraries. http://b/191699352 --- gpu.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gpu.Dockerfile b/gpu.Dockerfile index a6d6af2c..191d54a1 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -56,7 +56,7 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \ # the remaining pip commands: https://www.anaconda.com/using-pip-in-a-conda-environment/ # However, because this image is based on the CPU image, this isn't possible but better # to put them at the top of this file to minize conflicts. -RUN conda install cudf=21.06 cuml=21.06 cudatoolkit=$CUDA_VERSION && \ +RUN conda install cudf=21.08 cuml=21.08 cudatoolkit=$CUDA_VERSION && \ /tmp/clean-layer.sh # Install Pytorch and torchvision with GPU support.