Merge e7e197d into b260776

pytorch · Jul 3, 2023 · ed513d4 · ed513d4
2 parents b260776 + e7e197d
commit ed513d4
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"