制作Singularity镜像 深度学习+Anaconda3+tensorflow-gpu+pytorch

制作Singularity镜像 深度学习+Anaconda3+tensorflow-gpu+pytorch
首先写一个Singularity清单文件

mkdir test
cd test
vim dl.def

Singularity recipe内容如下：

Bootstrap: docker
From:nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04

%post
	apt-get update
	apt-get install -y wget \
		vim
	wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh -O /home/anaconda3.sh
	bash /home/anaconda3.sh -b -p /home/anaconda3
	rm /home/anaconda3.sh
	/home/anaconda3/bin/pip install --upgrade pip 
	/home/anaconda3/bin/pip install tensorflow-gpu
	/home/anaconda3/bin/pip install torch torchvision
	
%environment
	export PATH=/home/anaconda3/bin:$PATH

然后执行

sudo singularity build dl.simg dl.def

执行时，由于是gpu环境，需要加上命令–nv，例如：

singularity shell --nv dl.simg