按照官方文档安装NVIDIA Container Toolkit,一路下来都还比较顺利。
但是,不幸还是发生了,运行如下代码:
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
报错如下:
放狗各种搜索,其实是GFW的问题,解决方法如下:
sudo gedit /etc/docker/daemon.json
添加如下内容:
"registry-mirrors":["https://registry.docker-cn.com","http://hub-mirror.c.163.com","https://mirror.ccs.tencentyun.com","https://docker.1panel.live"]
完整json文档如下:
{"runtimes": {"nvidia": {"args": [],"path": "nvidia-container-runtime"}},"registry-mirrors":["https://registry.docker-cn.com","http://hub-mirror.c.163.com","https://mirror.ccs.tencentyun.com","https://docker.1panel.live"]
}
然后运行:
systemctl daemon-reload
systemctl restart docker
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
熟悉的画面出现了:
感叹一下,墙内搞点东西真不容易啊,一个破问题,一个小时没了。