From 83c7a598325e9d64cacdbf40daf4bcf60035610b Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Tue, 30 Apr 2024 19:37:59 -0400 Subject: [PATCH] Updates Docker image to fix MNIST example (#1213) Signed-off-by: Thomas J. Fan --- examples/mnist_classifier/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mnist_classifier/Dockerfile b/examples/mnist_classifier/Dockerfile index d8466061a..79ddd4179 100644 --- a/examples/mnist_classifier/Dockerfile +++ b/examples/mnist_classifier/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime +FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime LABEL org.opencontainers.image.source https://github.com/flyteorg/flytesnacks WORKDIR /root