温馨提示×

centos怎么下载pytorch

小樊
36
2026-10-23 23:44:14
栏目: 智能运维

在 CentOS 上下载/安装 PyTorch,通常有两种常用方式:pip 和 conda。下面按常见场景说明。


一、使用 pip 安装(推荐,简单)

1️⃣ 安装 Python 和 pip

CentOS 7/8 一般自带 Python,但建议用新版本:

# CentOS 7
sudo yum install -y python3 python3-pip

# 检查
python3 --version
pip3 --version

2️⃣ 安装 PyTorch(CPU 版本)

pip3 install torch torchvision torchaudio

3️⃣ 安装 CUDA 版本(如有 NVIDIA 显卡)

先确认 CUDA 版本:

nvidia-smi

例如 CUDA 11.8:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

二、使用 conda 安装(适合深度学习环境)

1️⃣ 安装 Miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
source ~/.bashrc

2️⃣ 创建环境并安装 PyTorch

conda create -n torch python=3.10
conda activate torch

CPU:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU(CUDA 11.8):

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

三、验证是否安装成功

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

四、常见问题

  • ❌ 网络慢:使用清华源
pip3 install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
  • ❌ CentOS 旧版 glibc 过低:建议使用 conda

如果你能告诉我:

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

我可以给你更精确的命令。

0 踩