correcting cuda 11.2 image name in CI and CD (apache#19960)

Co-authored-by: Rohit Kumar Srivastava <srivastava.141@buckeyemail.osu.edu>
access2rohit · Mar 10, 2021 · 45e6249 · 45e6249
1 parent 264bd8a
commit 45e6249
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/docker/Dockerfile.build.ubuntu_gpu_cu112 b/ci/docker/Dockerfile.build.ubuntu_gpu_cu112
@@ -18,7 +18,7 @@
 #
 # Dockerfile to run MXNet on Ubuntu 16.04 for GPU
 
-FROM nvidia/cuda:11.2-cudnn8-devel-ubuntu16.04
+FROM nvidia/cuda:11.2.1-cudnn8-devel-ubuntu16.04
 
 WORKDIR /work/deps