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