借助 docker 方式安装的很多大型项目都需要使用 nvidia 显卡,在 docker 中使用 nvidia 显卡资源,一方面需要在 host 主机上安装 nvidia 驱动,另外还需要安装 nvidia-docker。整个系统架构如下图所示

nvidia-docker.png (图片来源 https://github.com/NVIDIA/nvidia-docker )
本文总结了在 ubuntu 16.04 系统中 nvidia-docker 1.0 和 2.0 两个版本的安装方式。根据项目的不同要求,选择相应版本安装即可。
安装 nvidia-docker 1.0
- 添加 repo
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/

本文详细介绍了在Ubuntu16.04系统中通过Docker使用NVIDIA GPU的方法,包括nvidia-docker1.0和2.0的安装步骤,适用于不同项目需求。
最低0.47元/天 解锁文章
1798

被折叠的 条评论
为什么被折叠?



