Replace base of cuda_12 dockerfile with cudnn support one (#548)

tensorflow · Oct 6, 2023 · 8fb7907 · 8fb7907
1 parent a131e70
commit 8fb7907
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/perfzero/docker/Dockerfile_ubuntu_2204_tf_cuda_12 b/perfzero/docker/Dockerfile_ubuntu_2204_tf_cuda_12
@@ -2,7 +2,7 @@
 #  - Installs tf-nightly-gpu (this is TF 2.3)
 #  - Installs requirements.txt for tensorflow/models
 
-FROM nvidia/cuda:12.2.0-devel-ubuntu20.04 as base
+FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04 as base
 ARG tensorflow_pip_spec="tf-nightly"
 ARG local_tensorflow_pip_spec=""
 ARG extra_pip_specs=""