Skip to content
This repository was archived by the owner on Aug 7, 2025. It is now read-only.

Reduce TorchServe Docker GPU Image Size #2392

Merged
merged 10 commits into from
Jun 13, 2023
Merged

Reduce TorchServe Docker GPU Image Size #2392

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
0e5b4b6
Change nvidia base image to base from runtime
agunapal Jun 5, 2023
4c16169
Optimized Dockerfile
agunapal Jun 6, 2023
0b76950
Optimized Dockerfile
agunapal Jun 6, 2023
b3106d3
Merge branch 'master' into issues/reduce_docker_gpu_size
agunapal Jun 6, 2023
6b46425
Merge branch 'master' into issues/reduce_docker_gpu_size
agunapal Jun 12, 2023
75b9800
Changed to nvidia bas image
agunapal Jun 12, 2023
8203b2e
Merge branch 'issues/reduce_docker_gpu_size' of https://github.com/py…
agunapal Jun 12, 2023
dbe09af
Merge branch 'master' into issues/reduce_docker_gpu_size
agunapal Jun 12, 2023
1abe872
Merge branch 'master' into issues/reduce_docker_gpu_size
agunapal Jun 13, 2023
21da71b
Merge branch 'master' into issues/reduce_docker_gpu_size
agunapal Jun 13, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions docker/build_image.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ do
-g|--gpu)
MACHINE=gpu
DOCKER_TAG="pytorch/torchserve:latest-gpu"
BASE_IMAGE="nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu20.04"
BASE_IMAGE="nvidia/cuda:11.7.0-base-ubuntu20.04"
CUDA_VERSION="cu117"
shift
;;
Expand Down Expand Up @@ -82,10 +82,10 @@ do
CUDA_VERSION="$2"
if [ "${CUDA_VERSION}" == "cu118" ];
then
BASE_IMAGE="nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04"
BASE_IMAGE="nvidia/cuda:11.8.0-base-ubuntu20.04"
elif [ "${CUDA_VERSION}" == "cu117" ];
then
BASE_IMAGE="nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu20.04"
BASE_IMAGE="nvidia/cuda:11.7.0-base-ubuntu20.04"
elif [ "${CUDA_VERSION}" == "cu116" ];
then
BASE_IMAGE="nvidia/cuda:11.6.0-cudnn8-runtime-ubuntu20.04"
Expand Down