机器学习 Step By Step：Linux 配置 Nvidia Docker 以支持 Nvidia GPU

1. 安装 Nvidia 相关依赖

CentOS7

yum install -y nvidia-container-toolkit nvidia-container-runtime

2. 配置 dockerd 启动命令

修改 Systemd Service 配置文件

mkdir -p /etc/systemd/system/docker.service.d
tee /etc/systemd/system/docker.service.d/override.conf <<EOF
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd --host=fd:// --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
EOF
systemctl daemon-reload
systemctl restart docker

修改 Docker Daemon configuration file

tee