From 659fff35ed4db0d4d627793d64430474d4f02586 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Date: Thu, 2 Nov 2023 17:34:43 -0700
Subject: [PATCH 1/2] Update libtorch/Dockerfile to use Ubuntu-20.04

As 18.04 EOLed
---
 libtorch/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile
index 5d5b707af..8d69ac944 100644
--- 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

From e9a18f207afbe0d50107f2596181b2af06d5c751 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Date: Thu, 2 Nov 2023 17:38:21 -0700
Subject: [PATCH 2/2] Update build_docker.sh

---
 libtorch/build_docker.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libtorch/build_docker.sh b/libtorch/build_docker.sh
index c799bb178..8997f69cf 100755
--- 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)