温馨提示×

centos如何导入pytorch

小樊
49
2026-10-09 08:10:12
栏目: 智能运维

在 CentOS 上“导入 PyTorch”通常指两件事之一:

  1. 安装 PyTorch(让系统有 torch 可用)
  2. 在 Python 中 import torch(真正使用)

下面按标准流程说明。


一、准备 Python 环境(推荐)

CentOS 自带 Python 版本通常较旧,强烈建议使用虚拟环境。

1. 安装依赖

sudo yum install -y python3 python3-pip python3-devel gcc gcc-c++ make

2. 创建虚拟环境(推荐)

python3 -m venv pytorch_env
source pytorch_env/bin/activate

二、安装 PyTorch

✅ CPU 版本(最常见)

pip install torch torchvision torchaudio

✅ GPU 版本(需 NVIDIA 驱动 + CUDA)

先确认 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 特有)

1. GLIBC 版本过低

CentOS 7 可能报错:

ImportError: /lib64/libc.so.6: version `GLIBC_2.XX' not found

✅ 解决:

  • 使用 conda
  • 或升级系统(CentOS 8+)

2. 推荐使用 Conda(最省心)

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

如果你告诉我:

  • CentOS 版本(7 / 8 / Stream)
  • 是否用 GPU
  • 是否用 Conda

我可以给你完全匹配的一条命令。

0 踩