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