第一阶段:准备环境
首先需要安装一个工具包
pip install conda-pack
把主机的虚拟环境打包导出
conda pack -n 虚拟环境名字 -o 文件名.tar.gz
把本地的项目文件和这个压缩包放在一起,等下丢进去,现在让我们找一个基础镜像
docker pull meadml/cuda10.1-cudnn7-devel-ubuntu18.04-python3.6
这个是含有cuda和cudnn的基础镜像,但是没有anaconda,我们需要一个anaconda的离线安装包,我用的anaconda3-5.2.0.sh也和上面的文件放一起。
等镜像迁下来之后登进去。
docker run -it --gpus all meadml/cuda10.1-cudnn7-devel-ubuntu18.04-python3.6
会获取到一个容器短id,root@xxxxxxxxxxxx 那个xxxx就是。
把之前的文件拷贝进去。
docker cp 本地文件夹 容器id:/home/
进入home里面安装刚才复制进去的anaconda。
安装好了之后
source ~/.bashrc
就可以使用conda命令了。如果一切正确的话 nvcc -V命令和conda命令都会生效。
anaconda的安装目录会在/root/anaconda3这个地方,我们要报之前打包的虚拟环境.tar.gz导入进去。
cd /root/anaconda3/envs/
mkdir 你的虚拟环境名字
tar -xzf 你的虚拟环境.tar.gz -C /root/anaconda3/envs/你的虚拟环境名字/
解压进去就OK了