From 90929fa4333e5136020e9f9dcb7c1133e4c290f3 Mon Sep 17 00:00:00 2001 From: Jeff Yang Date: Tue, 6 Oct 2020 09:48:14 +0630 Subject: [PATCH] Fix apt repo issue for docker (#3823) * fix docker repo issue * docker * docker * docker * no cudnn * no cudnn * try 16.04 Co-authored-by: Jirka Borovec --- dockers/base-cuda/Dockerfile | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile index 2090320f3c..390d51ceb4 100644 --- a/dockers/base-cuda/Dockerfile +++ b/dockers/base-cuda/Dockerfile @@ -22,8 +22,11 @@ ARG CUDNN_VERSION=7 ARG CUDA_VERSION=10.1 -FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel -# FROM nvidia/cuda:${CUDA_VERSION}-devel +# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04 +# FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu18.04 +FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu16.04 +# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu18.04 +# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu16.04 ARG PYTHON_VERSION=3.7 ARG PYTORCH_VERSION=1.6