From e592710c25e3b029df61845c7e6f724c00d6236d Mon Sep 17 00:00:00 2001
From: Jirka Borovec <jirka@pytorchlightning.ai>
Date: Tue, 6 Oct 2020 10:58:50 +0200
Subject: [PATCH] use fine Ubuntu

---
 dockers/base-cuda/Dockerfile | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile
index 390d51ceb41e4..30c4c04c2bf7a 100644
--- a/dockers/base-cuda/Dockerfile
+++ b/dockers/base-cuda/Dockerfile
@@ -22,11 +22,9 @@
 ARG CUDNN_VERSION=7
 ARG CUDA_VERSION=10.1
 
-# FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
+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
@@ -37,7 +35,8 @@ SHELL ["/bin/bash", "-c"]
 
 ENV PATH="$PATH:/root/.local/bin"
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
+# todo: add back `apt-get update` when NVIDIA image is fixed
+RUN apt-get install -y --no-install-recommends \
         build-essential \
         cmake \
         git \