docker

先執行
sudo nvidia-ctk runtime configure –runtime=docker
會產生
/etc/docker/daemon.json
然後在倒數第2行加入
},
“default-runtime”: “nvidia”
}
然後
sudo systemctl restart docker