From 246b58eabf45f0aa25c296ddb8f2251465a0e3f0 Mon Sep 17 00:00:00 2001 From: Kaustav Mukherjee Date: Tue, 21 Jan 2020 05:09:59 +0530 Subject: [PATCH] Updated the Docker Image with the Latest one that uses GPU as of PR #255 Updated the docker image from pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime to kubeflow/pytorch:1.0-cuda10.0-cudnn7-runtime as the pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime is not GPU compatible.Hence using the Docker Image of PR #255. --- examples/mnist/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mnist/Dockerfile b/examples/mnist/Dockerfile index b4e02ced1..7e228b7f9 100644 --- a/examples/mnist/Dockerfile +++ b/examples/mnist/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime +FROM kubeflow/pytorch:1.0-cuda10.0-cudnn7-runtime RUN pip install tensorboardX==1.6.0 WORKDIR /var