Nvidia Docker安装后,使用镜像创建容器时出错,命令如下
docker run -it -gpus all ubuntu nvidia-smi
错误提示:
Error response from daemon: could not select device driver “” with
capabilities: [[gpu]]
解决:
安装NVIDIA Container Runtime
vi nvidia-container-runtime-script.sh
加入以下内容
sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
执行脚本
sh nvidia-container-runtime-script.sh
显示:
OK deb
https://nvidia.github.io/libnvidia-container/ubuntu18.04/ ( A R C H ) / d e b h t t p s : / / n v i d i a . g i t h u b . i o / n v i d i a − c o n t a i n e r − r u n t i m e / u b u n t u 18.04 / (ARCH) / deb https://nvidia.github.io/nvidia-container-runtime/ubuntu18.04/ (ARCH)/debhttps://nvidia.github.io/nvidia−container−runtime/ubuntu18.04/(ARCH)
/ Hit:1 http://archive.canonical.com/ubuntu bionic InRelease Get:2
https://nvidia.github.io/libnvidia-container/ubuntu18.04/amd64
InRelease [1139 B] Get:3
https://nvidia.github.io/nvidia-container-runtime/ubuntu18.04/amd64
InRelease [1136 B] Hit:4 http://security.ubuntu.com/ubuntu
bionic-security InRelease Get:5
https://nvidia.github.io/libnvidia-container/ubuntu18.04/amd64
Packages [4076 B] Get:6
https://nvidia.github.io/nvidia-container-runtime/ubuntu18.04/amd64
Packages [3084 B] Hit:7
http://us-east4-c.gce.clouds.archive.ubuntu.com/ubuntu bionic
InRelease Hit:8 http://us-east4-c.gce.clouds.archive.ubuntu.com/ubuntu
bionic-updates InRelease Hit:9
http://us-east4-c.gce.clouds.archive.ubuntu.com/ubuntu
bionic-backports InRelease Fetched 9435 B in 1s (17.8 kB/s)
Reading package lists… Done
安装:
$ apt-get install nvidia-container-runtime
使用 nvidia-ctk 命令配置容器运行时:
sudo nvidia-ctk runtime configure --runtime=docker
安装后重新启动docker
systemctl restart docker