快速制作深度学习docker镜像

第一步，安装docker(笔者为Centos7)

使用YUM 安装 docker_小前端而已的博客-优快云博客_yum安装docker

第二步，拉取deepo镜像

docker pull ufoym/deepo:all-jupyter-py36-cu100 

第三步，使用docker运行镜像并安装深度学习所需包

#运行python容器
docker run -it ufoym/deepo:all-jupyter-py36-cu100 /bin/bash

#安装lszrz
wget https://ohse.de/uwe/releases/lrzsz-0.12.20.tar.gz
tar -xvf lrzsz-0.12.20.tar.gz -C /usr/local/src/
cd /usr/local/src/lrzsz-0.12.20
./configure
make && make install
cd /usr/bin
ln -s /usr/local/bin/lrz rz
ln -s /usr/local/bin/lsz sz

#上传requirements.txt并安装
rz -y
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install --upgrade pip
pip install -r requirements.txt 

如果安装torch报错MemoryError

pip install --no-cache-dir torch==1.10.1

python导入torch报错MemoryError_掉坑也要弧线的博客-优快云博客

然后新建终端连接linux虚拟机

保存镜像文件

docker commit 7ee60fdf5128 torch-1.10.1:v1

docker save torch-1.10.1:v1 -o torch-1.10.1.images

#或者保存为tar
docker save torch-1.10.1:v1 > pytorch_1.10.1.tar