From b62800e5f586d0452dc0daf443f9324d38e12f95 Mon Sep 17 00:00:00 2001
From: Jayanth Kumar N <jayanth.kumar@students.iiserpune.ac.in>
Date: Mon, 4 Apr 2022 23:28:10 +0100
Subject: [PATCH] Updated CUDA and base OS version (#3060)

* Updated CUDA and base OS version

* Updated the MakeFile

To fix the issue of latest PyTorch version requiring CUDA > 9
---
 docker/Makefile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/Makefile b/docker/Makefile
index 90076eb75a..03051eaa82 100644
--- a/docker/Makefile
+++ b/docker/Makefile
@@ -1,8 +1,8 @@
 .PHONY: help create-host-workspace build build-gpu run run-gpu notebook notebook-gpu
 
 DOCKER_FILE=Dockerfile
-BASE_IMG=ubuntu:16.04
-BASE_CUDA_IMG=nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04
+BASE_IMG=ubuntu:18.04
+BASE_CUDA_IMG=nvidia/cuda:11.5.0-cudnn8-runtime-ubuntu18.04
 DOCKER_CMD=docker
 DOCKER_GPU_CMD=nvidia-docker
 HOST_WORK_DIR=${HOME}/pyro_docker