Change base image to smaller version (#330)

ml6team · Aug 1, 2023 · 4161b19 · 4161b19
1 parent f1ce3d3
commit 4161b19
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/components/segment_images/Dockerfile b/components/segment_images/Dockerfile
@@ -1,4 +1,4 @@
-FROM --platform=linux/amd64 pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
+FROM --platform=linux/amd64 pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 # System dependencies
 RUN apt-get update && \

diff --git a/docs/custom_component.md b/docs/custom_component.md
@@ -91,7 +91,7 @@ which is then imported in the `main.py` script.
 The `Dockerfile` defines how to build the component into a Docker image. An example Dockerfile is defined below.
 
 ```bash
-FROM --platform=linux/amd64 pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
+FROM --platform=linux/amd64 pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 ## System dependencies
 RUN apt-get update && \

diff --git a/scripts/build_components.sh b/scripts/build_components.sh
@@ -99,9 +99,6 @@ for dir in "${components_to_build[@]}"; do
    --label org.opencontainers.image.source=https://github.com/${namespace}/${repo} \
    .
 
-  # remove local image to conserve space 
-  docker rmi -f "${full_image_names[@]}" 
-
   popd
 
 done