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

最新推荐文章于 2025-06-20 11:02:34 发布

原创最新推荐文章于 2025-06-20 11:02:34 发布 · 3.4k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#Singularity #gpu #tensorflow #pytorch

深度学习专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何创建包含Anaconda3、TensorFlow-GPU和PyTorch的Singularity镜像，适用于深度学习GPU环境。步骤包括编写Singularity清单文件、安装依赖软件及构建和运行镜像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

制作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

然后执行