nvidia-docker – Tarik Billa

docker build with nvidia runtime

November 30, 2023 by Tarik

You need use nvidia-container-runtime as explained in docs: “It is also the only way to have GPU access during docker build”. Steps for Ubuntu: Install nvidia-container-runtime: sudo apt-get install nvidia-container-runtime Edit/create the /etc/docker/daemon.json with content: { “runtimes”: { “nvidia”: { “path”: “/usr/bin/nvidia-container-runtime”, “runtimeArgs”: [] } }, “default-runtime”: “nvidia” } Restart docker daemon: sudo systemctl restart … Read more

How do I specify nvidia runtime from docker-compose.yml?

August 21, 2023 by Tarik

Currently (Aug 2018), NVIDIA container runtime for Docker (nvidia-docker2) supports Docker Compose. Yes, use Compose format 2.3 and add runtime: nvidia to your GPU service. Docker Compose must be version 1.19.0 or higher. Example docker-compose.yml: version: ‘2.3’ services: nvsmi: image: ubuntu:16.04 runtime: nvidia environment: – NVIDIA_VISIBLE_DEVICES=all command: nvidia-smi More example from NVIDIA blog uses Docker … Read more