From b5497217b1c773a441e32193ec7bc23af815715e Mon Sep 17 00:00:00 2001 From: Thomas Eiszler Date: Mon, 28 Oct 2019 15:16:49 -0400 Subject: [PATCH] Basing off of CUDA10 to enable pyTorch 1.3 support. --- server/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/Dockerfile b/server/Dockerfile index 49c7c5f6..898e5725 100644 --- a/server/Dockerfile +++ b/server/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:8.0-cudnn5-devel +FROM nvidia/cuda:10.1-cudnn7-devel MAINTAINER Satyalab, satya-group@lists.andrew.cmu.edu ARG DEBIAN_FRONTEND=noninteractive