环境
Centos7.7,docker 19.03,显卡驱动:450.57
docker 19.03上,不再需要nvidia-docker2 工具包,因为nvidia gpu已经能够天然在docker 运行时支持。
如果已经使用了nvidia-docker2工具包,需要做特殊处理,网址:
https://github.com/NVIDIA/nvidia-docker/blob/master/README.md#quickstart
配置
生成脚本文件,内容如下:
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
sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker
其实这个脚本里就是添加源路径,然后下载安装nvidia-container-toolkit
重启docker 服务。
最后,将cuda的运行时库libcuda.so和libcudart.so 拷贝到容器系统中的/usr/lib64下。