centos7篇—安装nvidia-docker
前提:安装完docker,cuda,cudnn
具体安装过程可以看我的另两篇博客
禁用默认的驱动nouveau,安装cuda和cudnn
centos安装docker
当出现以下报错,查了很多,后来尝试安装 nvidia-smi,解决问题
安装 nvidia-docker
参考官方文档
Installation Guide
具体操作步骤为:
1. Setup the repository and the GPG key
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
> && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.repo | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo
2. 安装nvidia-docker2(网络网络不是很好,会稍微慢一点)
sudo yum install -y nvidia-docker2
3. 重启docker
sudo systemctl restart docker
4. 验证gpu是否可用
docker run --name py_11.3 --runtime=nvidia -it py_11.3:latest /bin/bash
报错成功解决,且GPU配置成功!!