docker容器内运行依赖GPU的程序

我的开发环境

1. centos7
2. docker版本为2.10.2
3. 2080ti显卡，物理机已安装cuda10和对应显卡驱动

运行带GPU的docker

1. 给linux安装nvidia-container-toolkit

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

2. 带gpu方式启动镜像 docker run -it —gpus all 766b2684d004

注意事项

1. 物理机器一定要安装cuda和驱动；
2. 容器无需安装显卡驱动，但要安装cuda且兼容物理机显卡驱动；
3. 容器配置cuda后，可用命令nvidia-smi检测；
4. cuda、显卡驱动安装，可查看我其他博客教程；