简单总结一下如何搭 Nvidia Docker

为了用 TensorFlow 的一些新特性,需要 CUDA9,但 Debian9 死活都装不上 CUDA9……

那只好搭个 Docker 试试了

写在前面:Docker 依赖宿主机的 Nvidia 驱动……所以原始设想“Docker 里面用 CUDA9、宿主机用 CUDA8”是不太可能了……

安装 Docker CE

这里 ,找到对应的平台(比如 Debian),安装依赖和 Docker

其实可以仔细找一下,直接下载 deb 安装包进行安装的……

或者官方有一个一键脚本: curl -sSL https://get.docker.com | sh

安装 Nvidia 驱动

这个…… 略吧…… 个人不喜欢使用 run-file 进行安装。使用 deb 包多简单啊,而且每次有更新都会自动编译内核扩展。

安装 Nvidia Docker

这里 ,找到对应的架构,进行安装

然后

最后测试一下

不想每次都加 sudo

然后登出再登入。建议是将所有需要用 Docker 的人都加入到这个组。

备选方案在 这里

国内镜像源

鉴于国内网络现状…… 请 参考这篇文章

写在最后

什么鬼!!!Docker 里面的 CUDA 是依赖宿主机的 CUDA 的!!!

也就是说,想要“宿主 CUDA8,Docker CUDA9”是不可能的!!!

好了安心看贝叶斯去了……

参考资料

发表评论

电子邮件地址不会被公开。 必填项已用*标注