使用 Docker 搭建深度学习训练环境

使用 Docker 可以搭建自己的深度学习训练环境,以避免不同训练因需要的运行环境不同而导致冲突。

此环境使用 NVIDIA 的 cuda 计算框架,因此构建环境以 NVIDIA 提供的镜像为基础。

由于目前 NVIDIA 官方并未公布适用于 ubuntu 18.04 的计算框架,因此现在以 ubuntu 16.04 为基础。待适用于 ubuntu 18.04 的计算框架全部发布以后再构建适用于 ubuntu 18.04 的版本。

构建环境

1.构建基础环境(Dockerfile_9.2-base-1604)

编写 Dockerfile:

其中 maintainer 标签可以改为你自己的信息,此后不再说明。

 

发表评论

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据