简单总结一下如何搭Nvidia Docker

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

那只好搭个 Docker 试试了

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

安装 Docker CE

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

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

安装 Nvidia 驱动

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

安装 Nvidia Docker

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

然后

最后测试一下

不想每次都加 sudo

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

备选方案在 这里

国内镜像源

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

写在最后

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

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

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

参考资料

发表评论

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