fix: switch to nightly for cve fix

bahe-msft · Nov 19, 2024 · d940635 · d940635
1 parent 07b315a
commit d940635
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mnist/Dockerfile b/mnist/Dockerfile
@@ -1,4 +1,5 @@
-FROM tensorflow/tensorflow:2.18.0-gpu
+# tf nightly-gpu: https://hub.docker.com/layers/tensorflow/tensorflow/nightly-gpu/images/sha256-31e8f7fa6e7abee03c407deb776d272bac884b9f773e68161aae02d3a22932cb?context=explore
+FROM tensorflow/tensorflow@sha256:31e8f7fa6e7abee03c407deb776d272bac884b9f773e68161aae02d3a22932cb
 
 ADD model.py /opt/model.py