Updated to nvidia/cuda:11.7.1-base-ubuntu20.04 (#2442)

pytorch · Jul 6, 2023 · 4d7dc64 · 4d7dc64
1 parent b260776
commit 4d7dc64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/build_image.sh b/docker/build_image.sh
@@ -43,7 +43,7 @@ do
         -g|--gpu)
           MACHINE=gpu
           DOCKER_TAG="pytorch/torchserve:latest-gpu"
-          BASE_IMAGE="nvidia/cuda:11.7.0-base-ubuntu20.04"
+          BASE_IMAGE="nvidia/cuda:11.7.1-base-ubuntu20.04"
           CUDA_VERSION="cu117"
           shift
           ;;
@@ -85,7 +85,7 @@ do
             BASE_IMAGE="nvidia/cuda:11.8.0-base-ubuntu20.04"
           elif [ "${CUDA_VERSION}" == "cu117" ];
           then
-            BASE_IMAGE="nvidia/cuda:11.7.0-base-ubuntu20.04"
+            BASE_IMAGE="nvidia/cuda:11.7.1-base-ubuntu20.04"
           elif [ "${CUDA_VERSION}" == "cu116" ];
           then
             BASE_IMAGE="nvidia/cuda:11.6.0-cudnn8-runtime-ubuntu20.04"