centos如何迁移pytorch

在 CentOS 上“迁移 PyTorch”通常有两种理解，我先分别说明，你可以对号入座：

一、在同一台 CentOS 上迁移 PyTorch 环境（换版本 / 换环境）

1️⃣ 使用 conda 迁移（推荐）

如果你原来用 Anaconda / Miniconda：

# 查看已有环境
conda env list

# 导出原环境
conda activate old_env
conda env export > pytorch_env.yaml

# 在新环境或新机器创建
conda env create -f pytorch_env.yaml

✅ 优点：最完整、最不容易出错

2️⃣ 使用 pip 迁移（无 conda）

# 导出依赖
pip freeze > requirements.txt

# 新环境安装
pip install -r requirements.txt

⚠️ 注意：

• 需保证 CUDA / driver 版本一致
• torch / torchvision 版本要匹配

二、把 PyTorch 从一台 CentOS 迁移到另一台 CentOS

场景 A：整机 / 服务器迁移

如果是换服务器：

1. 原机器导出环境（见上面）
2. 新机器安装相同：
   • CentOS 版本（7 / 8 / Stream）
   • NVIDIA 驱动
   • CUDA / cuDNN
3. 恢复 Python 环境

场景 B：只迁移 PyTorch 模型 / 代码

如果只是模型和代码：

project/
├── model.pth
├── train.py
├── inference.py
└── requirements.txt

新机器：

pip install torch torchvision

三、CentOS 常见坑（很重要）

❗ 1. CUDA 版本不兼容

nvidia-smi        # 驱动支持的最高 CUDA
nvcc --version    # 当前 CUDA

PyTorch 必须匹配：

pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html

❗ 2. glibc / gcc 太旧（CentOS 7）

CentOS 7 默认：

• gcc 4.8
• glibc 2.17

✅ 解决方案：

# 使用 conda 自带编译器
conda install -c conda-forge gcc_linux-64 gxx_linux-64

❗ 3. 离线迁移

如果新机器 不能联网：

# 原机器下载
pip download torch -d ./wheels

# 新机器安装
pip install --no-index --find-links=./wheels torch

四、快速检查是否迁移成功

import torch
print(torch.__version__)
print(torch.cuda.is_available())

如果你愿意，可以告诉我：

• ✅ CentOS 版本（7 / 8 / Stream）
• ✅ 是否用 GPU
• ✅ 原环境是 conda 还是 pip
• ✅ 是换机器还是换版本

我可以给你一步步精确命令。