编程语言
首页 > 编程语言> > 使用singularity制作python镜像在服务器上训练数据

使用singularity制作python镜像在服务器上训练数据

作者:互联网

创建python镜像

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

镜像文件在训练文件的上一层文件夹,接下来就可以把这个镜像扔到服务器上开始训练数据了,不需要在考虑环境的安装问题。

标签:cn,python,pytorch,sif,镜像,singularity
来源: https://www.cnblogs.com/lisuhang/p/16159475.html