温馨提示×

centos pytorch怎么备份

小樊
46
2026-09-24 16:30:25
栏目: 智能运维

在 CentOS 上备份 PyTorch 相关环境,一般有几种层面,你可以根据需求选择:


一、备份 PyTorch 的 Python 环境(最常用 ✅)

1️⃣ 使用 conda 备份(如果你用 Anaconda / Miniconda)

conda env export > pytorch_env.yaml

恢复:

conda env create -f pytorch_env.yaml

✅ 推荐,最完整(包含 CUDA、cuDNN、PyTorch 版本)


2️⃣ 使用 pip 备份(虚拟环境或系统 Python)

pip freeze > requirements.txt

恢复:

pip install -r requirements.txt

⚠️ 注意:

  • pip 不会记录 CUDA 版本
  • 适合 CPU / 固定 CUDA 环境

二、备份整个 Conda 环境目录(最稳妥 ✅✅)

cp -r ~/miniconda3/envs/pytorch_env /backup/pytorch_env_backup

恢复:

cp -r /backup/pytorch_env_backup ~/miniconda3/envs/pytorch_env

✅ 不依赖网络
❌ 占用空间大


三、备份 PyTorch 模型与数据

1️⃣ 模型权重

cp -r ./checkpoints /backup/pytorch_checkpoints

2️⃣ 数据集

tar -czf data_backup.tar.gz ./dataset

四、备份系统级 PyTorch(不常见)

如果你是用 yum 或源码安装:

rpm -qa | grep torch

或备份:

/usr/local/lib/python3.x/site-packages/torch

⚠️ 不推荐,建议用虚拟环境


五、推荐方案(总结 ✅)

场景 推荐方式
日常实验 conda env export
服务器迁移 整个 env 目录拷贝
只备份代码 requirements.txt
模型安全 定期 tar 模型

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

  • 用的是 conda 还是 pip
  • 是否有 GPU / CUDA
  • 是 开发机还是生产服务器

我可以给你一套专属备份脚本。

0 踩