We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8682b1b commit 1ac609fCopy full SHA for 1ac609f
Dockerfile.tmpl
@@ -1,9 +1,9 @@
1
{{ if eq .Accelerator "gpu" }}
2
-FROM gcr.io/deeplearning-platform-release/tf2-gpu.2-6:m78
+FROM gcr.io/deeplearning-platform-release/tf2-gpu.2-6:m80
3
ENV CUDA_MAJOR_VERSION=11
4
ENV CUDA_MINOR_VERSION=0
5
{{ else }}
6
-FROM gcr.io/deeplearning-platform-release/tf2-cpu.2-6:m78
+FROM gcr.io/deeplearning-platform-release/tf2-cpu.2-6:m80
7
{{ end }}
8
# Keep these variables in sync if base image is updated.
9
ENV TENSORFLOW_VERSION=2.6.0
0 commit comments