之前一直用着原生系统搭建起来的环境,用着还算稳定,不过偶尔也会遇到一些困扰。比如tf,pytorch这类框架的API不是很稳定,版本更新后老的代码用不了。有时候跑别人的程序需要切换对应的版本。
在师弟的安利下尝试了一下用docker来运行深度学习环境。
之前用过几次,但是对docker的了解较少,稍微做了一点功课,Docker — 从入门到实践。
这里略过le docker的安装,详见docker的官方文档。
安装nvidia-docker
对于深度学习开发环境来说,CUDA环境是至关重要的,要在docker中训练模型,首先得要能够在docker中运行CUDA,nvidia官方给出了解决方案————nvidia-docker。