Linux中利用docker搭建深度学习环境

写在前面：

在深度学习中，避免不了在远程服务器上进行模型的训练，如果直接在服务器裸机的基础环境跑显然是不可取的，此时搭建用于模型训练的docker环境显得尤为重要。

1.下载基础镜像

选择一个合适的基础镜像会给后续的操作带来极大的便利，其中ufoym/deepo镜像是笔者最常用的一个，输入如下命令下载基础镜像

docker pull ufoym/deepo

下载好后，输入如下命令，列表中若出现ufoym/deepo则下载成功

docker images

2.创建docker容器

根据下载好的镜像来创建容器,基本的格式为：docker run --runtime=nvidia -it --name 容器名称 --gpus all -v /服务器基础环境下的目录:/容器中的目录 ufoym/deepo bash

• -name 设置容器名称
• -gpus 指定GPU个数
• -it 交互式进入
• -v 设置文件映射路径

举例如下：

docker run --runtime=nvidia -it