docker-compose版本: 3.X
对于2.X版本,可以添加runtimes熟悉。参考:
添加
{
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
到 /etc/docker/daemon.json
然后使用 --runtime=nvidia
docker-compose:
runtime: nvidia
environment:
-NVIDIA_VISIBLE_DEVICES=all
**注意这是2.X 版本的docker-compose!!!**3.X版本没有/usr/bin/nvidia-container-runtime any more…
或者你可以安装 nvidia-docker-runtime:
https://github.com/NVIDIA/nvidia-container-runtime#docker-engine-setup
add to /etc/docker/daemon.json
{
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
docker-compose:
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
并没有尝试这么做,想等docker-compose更新对nvidia-gpu的支持。尝试用其它技术路径部署docker。
更新于2020.5.26.保质期不知道多久,不想误人子弟。