在 CentOS 上备份 PyTorch 相关环境,通常分几种情况。下面按“最常见、最推荐”到“系统级”给你梳理。
# 查看环境
conda env list
# 备份指定环境
conda env export -n pytorch_env > pytorch_env.yaml
恢复:
conda env create -f pytorch_env.yaml
✅ 优点
pip freeze > requirements.txt
恢复:
pip install -r requirements.txt
⚠️ 注意
python -c "import torch; print(torch.__version__, torch.version.cuda)"
cp -r ./checkpoints /backup/pytorch/checkpoints_$(date +%F)
tar czf pytorch_code_$(date +%F).tar.gz ./project
如果你是直接装在系统 Python:
# 查看安装位置
python -c "import torch; print(torch.__file__)"
备份:
cp -r /usr/lib64/python3.x/site-packages/torch /backup/
⚠️ 不推荐,恢复复杂、容易版本冲突
nvidia-smi
nvcc --version
CUDA 一般 不备份,重装更快
只记录版本号即可
#!/bin/bash
DATE=$(date +%F)
mkdir -p /backup/pytorch_$DATE
conda env export -n pytorch_env > /backup/pytorch_$DATE/env.yaml
cp -r ./checkpoints /backup/pytorch_$DATE/
tar czf /backup/pytorch_$DATE/code.tar.gz ./project
✅ 最佳实践
❌ 不推荐
如果你愿意,可以告诉我:
我可以给你定制备份方案。