diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index f6838975332b..d5dbb9115138 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:10.0-cudnn7-devel-ubuntu18.04 +FROM nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04 # Base scripts RUN apt-get update --fix-missing