You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Jul 10, 2025. It is now read-only.
Currently, the Docker install GPU support instructions here:
https://www.tensorflow.org/install/install_linux#gpu_support
Indicates it requires the full CUDA toolkit, cuDNN, and Nvidia drivers as outlined in:
https://www.tensorflow.org/install/install_linux#NVIDIARequirements
The document should be changed to indicate the host machine (running the container) requires only Nvidia drivers. See:
https://github.com/nvidia/nvidia-docker/wiki/CUDA#requirements
The container itself requires only the CUDA toolkit, cuDNN and Docker file can be found here:
https://hub.docker.com/r/nvidia/cuda/