在 CentOS 离线环境 安装 PyTorch,核心思路是:在有网络的机器上下载好安装包,再拷贝到离线机器安装。下面按常见场景一步步说明。
cat /etc/centos-release
uname -m
常见:
nvidia-smi
pip download torch torchvision torchaudio \
-d ./pytorch_packages \
--index-url https://download.pytorch.org/whl/cpu
如果是 CUDA 11.8:
pip download torch torchvision torchaudio \
-d ./pytorch_packages \
--index-url https://download.pytorch.org/whl/cu118
⚠️ 注意:
- 下载机器和离线机器 Python 版本要一致
- 系统架构要一致(x86_64)
访问: https://download.pytorch.org/whl/
手动下载:
scp -r pytorch_packages user@offline-server:/tmp/
或 U 盘 / 内网传输均可。
yum install -y python3 python3-pip
如果是 GPU:
yum install -y gcc gcc-c++
cd /tmp/pytorch_packages
pip3 install *.whl
或指定顺序(更稳妥):
pip3 install torch-*.whl
pip3 install torchvision-*.whl
pip3 install torchaudio-*.whl
python3 -c "import torch; print(torch.__version__)"
GPU 版本:
python3 -c "import torch; print(torch.cuda.is_available())"
在联网机器提前下载:
pip download numpy typing -d ./pytorch_packages
解决方案:
| 场景 | 版本 |
|---|---|
| CPU | PyTorch 2.1 + Python 3.9 |
| CUDA 11.8 | PyTorch 2.1 |
| CentOS 7 | PyTorch ≤ 2.0 |
如果你愿意,可以告诉我:
我可以给你 精确到命令的离线安装方案。