Update libtorch/Dockerfile to use Ubuntu-20.04 (#1578)

As 18.04 EOLed
pytorch · Nov 3, 2023 · df5f7c5 · df5f7c5
1 parent 9467b4e
commit df5f7c5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile
@@ -1,5 +1,5 @@
 ARG BASE_TARGET=base
-ARG GPU_IMAGE=ubuntu:18.04
+ARG GPU_IMAGE=ubuntu:20.04
 FROM ${GPU_IMAGE} as base
 
 ENV DEBIAN_FRONTEND=noninteractive

diff --git a/libtorch/build_docker.sh b/libtorch/build_docker.sh
@@ -15,13 +15,13 @@ case ${GPU_ARCH_TYPE} in
     cpu)
         BASE_TARGET=cpu
         DOCKER_TAG=cpu
-        GPU_IMAGE=ubuntu:18.04
+        GPU_IMAGE=ubuntu:20.04
         DOCKER_GPU_BUILD_ARG=""
         ;;
     cuda)
         BASE_TARGET=cuda${GPU_ARCH_VERSION}
         DOCKER_TAG=cuda${GPU_ARCH_VERSION}
-        GPU_IMAGE=ubuntu:18.04
+        GPU_IMAGE=ubuntu:20.04
         DOCKER_GPU_BUILD_ARG=""
         ;;
     rocm)