Nvidia Docker安装后,使用镜像创建容器时出错,命令如下

docker run -it -gpus all ubuntu nvidia-smi

错误提示:

Error response from daemon: could not select device driver "" with capabilities: [[gpu]]

这是由于没有配置好docker运行环境导致的,执行安装容器运行环境(Ubuntu为例)

apt-get install nvidia-container-toolkit

如果找不到,请添加

$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

$ sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
$ sudo systemctl restart docker

安装后重新启动docker

systemctl restart docker

更多推荐