创建或修改 /etc/docker/daemon.json 文件,修改为如下形式
{
" registry-mirrors" : [
" https://registry.docker-cn.com" ,
" https://docker.mirrors.ustc.edu.cn" ,
" http://hub-mirror.c.163.com"
]
}
添加用户到docker组并重启docker服务使配置生效
add $USER to docker-group
$ sudo usermod -aG docker $USER
$ sudo gpasswd -a $USER docker
$ newgrp docker
$ systemctl restart docker.service
$ distribution=$( . /etc/os-release; echo $ID$VERSION_ID )
$ curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution /libnvidia-container.list | \
sed ' s#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
$ sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
$ sudo nvidia-ctk runtime configure --runtime=docker
$ sudo systemctl restart docker
$ sudo docker run --rm --runtime=nvidia --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi