Ubuntu21.04无法安装nvidia-docker

原因：目前还没有nvidia-docker版本支持ubuntu21.04
报错：Unsupported distribution!
可查看支持的版本：https://nvidia.github.io/nvidia-docker/

解决方法：
添加存储库：/etc/apt/sources.list.d/nvidia-docker.list

vi /etc/apt/sources.list.d/nvidia-docker.list

把添加以下内容：
deb https://nvidia.github.io/libnvidia-container/ubuntu18.04/amd64 /
deb https://nvidia.github.io/nvidia-container-runtime/ubuntu18.04/amd64 /
deb https://nvidia.github.io/nvidia-docker/ubuntu18.04/amd64 /

#更新源

sudo apt-get update

#安装

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

#验证

docker run --runtime=nvidia --rm nvidia/cuda:11.0-base nvidia-smi#注意选择自己的cuda版本

ps:后续如果有nvidia-docker支持ubuntu21.04的版本，请按官方操作