Just use centos:7

Author: malfet

conda/Dockerfile

@@ -1,6 +1,6 @@
 ARG CUDA_VERSION=10.2
 ARG BASE_TARGET=cuda${CUDA_VERSION}
-FROM nvidia/cuda:11.4.3-devel-centos7 as base
+FROM centos:7 as base
 
 ENV LC_ALL en_US.UTF-8
 ENV LANG en_US.UTF-8

libtorch/Dockerfile

@@ -1,5 +1,5 @@
 ARG BASE_TARGET=base
-ARG GPU_IMAGE=nvidia/cuda:11.4.3-devel-ubuntu18.04
+ARG GPU_IMAGE=ubuntu:18.04
 FROM ${GPU_IMAGE} as base
 
 ENV DEBIAN_FRONTEND=noninteractive

libtorch/build_docker.sh

@@ -15,13 +15,13 @@ case ${GPU_ARCH_TYPE} in
     cpu)
         BASE_TARGET=cpu
         DOCKER_TAG=cpu
-        GPU_IMAGE=nvidia/cuda:11.4.3-devel-ubuntu18.04
+        GPU_IMAGE=ubuntu:18.04
         DOCKER_GPU_BUILD_ARG=""
         ;;
     cuda)
         BASE_TARGET=cuda${GPU_ARCH_VERSION}
         DOCKER_TAG=cuda${GPU_ARCH_VERSION}
-        GPU_IMAGE=nvidia/cuda:11.4.3-devel-ubuntu18.04
+        GPU_IMAGE=ubuntu:18.04
         DOCKER_GPU_BUILD_ARG=""
         ;;
     rocm)