singularity build --sandbox pytorch docker://python:3.8
创建一个名字叫做pytorch的镜像,python 版本选择3.8
singularity shell -w pytorch
通过shell进入交互模式,安装需要的python包
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
或者
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple packname
sudo singularity build pytorch.sif pytorch/
在创建镜像的根目录,用镜像构建sif文件
singularity exec --nv ../pytorch.sif python train.py
镜像文件在训练文件的上一层文件夹,接下来就可以把这个镜像扔到服务器上开始训练数据了,不需要在考虑环境的安装问题。