Skip to content
This repository has been archived by the owner on Dec 9, 2024. It is now read-only.

Commit

Permalink
Replace base of cuda_12 dockerfile with cudnn support one (#548)
Browse files Browse the repository at this point in the history
  • Loading branch information
@ortibazar
ortibazar authored Oct 6, 2023
1 parent a131e70 commit 8fb7907
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion perfzero/docker/Dockerfile_ubuntu_2204_tf_cuda_12
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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=""
Expand Down

0 comments on commit 8fb7907

Please sign in to comment.