5 ms·
At NVIDIA we maintain this utility: https://github.com/NVIDIA/nvidia-docker https://github.com/NVIDIA/nvidia-docker It automatically discovers the devices and
by flx42_ 10y ago
At NVIDIA we maintain this utility:
https://github.com/NVIDIA/nvidia-docker https://github.com/NVIDIA/nvidia-docker
It automatically discovers the devices and the right driver files on the host.
The main goal is compute (CUDA), but we also demonstrated how to run TF2 on Steam OS during our DockerCon 16 OpenForum presentation.
Nice job! :)
- voltagex_ 10y agoI'm really not up on how this all works, but isn't https://github.com/NVIDIA/nvidia-docker/blob/master/ubuntu-16.04/cuda/8.0/runtime/Dockerfile#L17 https://github.com/NVIDIA/nvidia-docker/blob/master/ubuntu-1... hardcoding driver versions in a different way?
- flx42_ 10y agoNo, this is the CUDA toolkit, it doesn't depend on the driver version. You can compile CUDA code without having a GPU (which is the case during a "docker build"). Edit: in other words, your Docker image doesn't depend on a specific driver version and can be ran on any machine with sufficient drivers. Driver files are mounted as a volume when starting the container.