diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index daf6381908d8..3efd81496a66 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -17,7 +17,7 @@ # CI docker GPU env # tag: v0.60 -FROM nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04 +FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu18.04 COPY utils/apt-install-and-clear.sh /usr/local/bin/apt-install-and-clear