sudo vi /etc/docker/daemon.json
{ "default-runtime": "nvidia", "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } }, }
sudo service docker restart
* you can try to print the nvidia-smi when building a docker image using Dockerfile
inside your build.sh, add the following command, DOCKER_BUILDKIT=0 :-
DOCKER_BUILDKIT=0 docker build . -t image_name
No comments:
Post a Comment