could not select device driver “nvidia“ with capabilities: [[gpu]]错误解决

1. centos系统

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

yum clean expire-cache

安装nvidia驱动

yum install -y nvidia-docker2

重启docker

sudo systemctl restart docker

2. ubuntu系统

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \   sudo apt-key add - 
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) 
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

安装nvidia-container-toolkit

sudo apt-get install -y nvidia-container-toolkit

重启docker

sudo systemctl restart docker