温馨提示×

centos pytorch如何备份

小樊
45
2026-09-15 14:38:16
栏目: 智能运维

CentOS 上备份 PyTorch 相关环境,通常分几种情况。下面按“最常见、最推荐”到“系统级”给你梳理。


一、最推荐:用 conda / pip 备份 PyTorch 环境(✅ 首选)

1️⃣ 如果是 conda 环境

# 查看环境
conda env list

# 备份指定环境
conda env export -n pytorch_env > pytorch_env.yaml

恢复:

conda env create -f pytorch_env.yaml

✅ 优点

  • 包含 PyTorch 版本、CUDA、依赖
  • 跨机器可恢复

2️⃣ 如果是 pip 环境

pip freeze > requirements.txt

恢复:

pip install -r requirements.txt

⚠️ 注意

  • pip 不会记录 CUDA 驱动
  • 建议同时记录:
python -c "import torch; print(torch.__version__, torch.version.cuda)"

二、备份 PyTorch 模型 & 训练结果(非常重要)

1️⃣ 模型权重

cp -r ./checkpoints /backup/pytorch/checkpoints_$(date +%F)

2️⃣ 训练代码

tar czf pytorch_code_$(date +%F).tar.gz ./project

三、备份系统级 PyTorch(不推荐但存在)

如果你是直接装在系统 Python:

# 查看安装位置
python -c "import torch; print(torch.__file__)"

备份:

cp -r /usr/lib64/python3.x/site-packages/torch /backup/

⚠️ 不推荐,恢复复杂、容易版本冲突


四、CUDA / 驱动备份(可选)

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

六、总结(建议做法)

最佳实践

  • conda / pip 导出环境
  • 定期备份模型 & 代码
  • 记录 CUDA / PyTorch 版本

不推荐

  • 直接拷系统 Python
  • 只备份 PyTorch 包

如果你愿意,可以告诉我:

  • CentOS 版本(7 / 8 / Stream)
  • PyTorch 安装方式(conda / pip / 源码)
  • 是否用 GPU

我可以给你定制备份方案

0