Updates Docker image to fix MNIST example (#1213)

Signed-off-by: Thomas J. Fan <[email protected]>
flyteorg · Apr 30, 2024 · 83c7a59 · 83c7a59
1 parent e89c659
commit 83c7a59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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