From f44a19663159b521a2f7209c12c40cfe3ead7bd9 Mon Sep 17 00:00:00 2001 From: VitorGuizilini-TRI <58576956+VitorGuizilini-TRI@users.noreply.github.com> Date: Mon, 1 Jun 2020 19:45:22 -0700 Subject: [PATCH] Fix dockerfile project (#22) --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index b58c06f4..25f726ee 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -2,7 +2,7 @@ FROM nvidia/cuda:10.1-devel-ubuntu18.04 -ENV PROJECT=packnet +ENV PROJECT=packnet-sfm ENV PYTORCH_VERSION=1.4.0 ENV TORCHVISION_VERSION=0.5.0 ENV CUDNN_VERSION=7.6.5.32-1+cuda10.1