在 CentOS 上将 PyTorch 环境“迁移”,通常有三种常见场景。下面按最常见需求给你讲清楚怎么做,你可以对号入座。
如果你是用 Anaconda / Miniconda 安装的 PyTorch:
conda env export > pytorch_env.yaml
conda env create -f pytorch_env.yaml
conda activate pytorch_env
⚠️ 注意:
pip freeze > requirements.txt
在新环境:
pip install -r requirements.txt
⚠️ pip 不会记录 CUDA / cuDNN,需要手动保证一致。
# 旧机器
tar -czf pytorch_env.tar.gz ~/miniconda3/envs/pytorch_env
# 新机器
tar -xzf pytorch_env.tar.gz -C ~/miniconda3/envs/
⚠️ 注意:
conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
或 pip:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
FROM nvidia/cuda:11.8-cudnn8-runtime-centos7
RUN yum install -y python3 python3-pip
RUN pip3 install torch torchvision torchaudio
好处:
❌ glibc 版本过低(CentOS 7 常见)
❌ CUDA 驱动版本不匹配
❌ 用 root 装环境,普通用户用不了
✅ 建议:
你可以告诉我:
我可以直接给你一条可复制的命令清单。