Basing off of CUDA10 to enable pyTorch 1.3 support.

cmusatyalab · Oct 28, 2019 · b549721 · b549721
1 parent 4bfe58f
commit b549721
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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, [email protected]
 
 ARG DEBIAN_FRONTEND=noninteractive