在 CentOS 上“导入 PyTorch”通常指两件事之一:
下面按标准流程说明。
CentOS 自带 Python 版本通常较旧,强烈建议使用虚拟环境。
sudo yum install -y python3 python3-pip python3-devel gcc gcc-c++ make
python3 -m venv pytorch_env
source pytorch_env/bin/activate
pip install torch torchvision torchaudio
先确认 CUDA 版本:
nvidia-smi
例如 CUDA 11.8:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
官方地址:https://pytorch.org/get-started/locally/
python
在 Python 中执行:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # GPU 时为 True
无报错即成功。
CentOS 7 可能报错:
ImportError: /lib64/libc.so.6: version `GLIBC_2.XX' not found
✅ 解决:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
conda create -n torch python=3.10
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
CentOS 导入 PyTorch = 安装 torch + import torch
如果你告诉我:
我可以给你完全匹配的一条命令。