服务器上配置pytorch-开发者社区

一、前置准备：检查服务器基础环境

首先通过 SSH 连接服务器，执行以下命令检查关键信息，确保安装适配：

1. 检查系统与 Python 版本

bash

# 查看系统版本（确认是Linux，如CentOS/Ubuntu） cat /etc/os-release # 查看Python版本（PyTorch要求Python 3.8+，推荐3.9/3.10） python3 --version # 或 python --version（部分服务器Python3已设为默认）

2. 检查 GPU 与 CUDA 版本（仅 GPU 版需要）

如果服务器有 NVIDIA 显卡，需确认 CUDA 版本（PyTorch 与 CUDA 版本需匹配）：

bash

# 查看显卡信息和驱动版本 nvidia-smi # 查看已安装的CUDA Toolkit版本（可选） nvcc -V

重点关注nvidia-smi输出中的CUDA Version（如 12.1、11.8），这是驱动支持的最高 CUDA 版本，安装 PyTorch 时需选择≤该版本的 CUDA。

二、环境准备：安装 Anaconda（推荐）

服务器多为多用户环境，使用 Anaconda 创建独立虚拟环境可避免依赖冲突，是 PyTorch 安装的最佳实践：

1. 下载并安装 Anaconda

bash

# 下载Anaconda（Linux x86_64版本，适配绝大多数服务器） wget https://repo.anaconda.com/archive/Anaconda3-2024.02-1-Linux-x86_64.sh # 执行安装脚本（一路回车，最后输入yes同意conda init） bash Anaconda3-2024.02-1-Linux-x86_64.sh # 刷新环境变量，使conda生效 source ~/.bashrc

2. 创建并激活 PyTorch 虚拟环境

bash

# 创建名为pytorch_env的环境，指定Python版本（如3.10） conda create -n pytorch_env python=3.10 -y # 激活环境（后续所有操作需在该环境下执行） conda activate pytorch_env

三、安装 PyTorch（分 CPU/GPU 版）

方式 1：GPU 版（优先，服务器有显卡时）

推荐通过 PyTorch 官方命令安装，自动匹配 CUDA 版本：

打开 PyTorch 官方安装页面
按服务器环境选择参数：
- OS：Linux
- Package：Conda（推荐）/ Pip
- Language：Python
- Compute Platform：匹配服务器的 CUDA 版本（如 CUDA 12.1、11.8）

复制官方生成的命令执行，例如（CUDA 11.8 + Conda）：

bash

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

方式 2：CPU 版（无显卡 / 仅轻量使用）

bash

# Conda安装（推荐） conda install pytorch torchvision torchaudio cpuonly -c pytorch # 或Pip安装 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

补充：手动指定版本安装（如需特定版本）

bash

# 例如安装PyTorch 2.1.0 + CUDA 11.8 conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia

四、验证 PyTorch 安装是否成功

激活虚拟环境后，执行以下 Python 代码验证：

bash

# 进入Python交互环境 python

在 Python 终端中输入：

python

import torch # 检查PyTorch版本 print("PyTorch版本：", torch.__version__) # 检查CUDA是否可用（GPU版需验证） print("CUDA是否可用：", torch.cuda.is_available()) # 检查可用GPU数量（GPU版） print("可用GPU数：", torch.cuda.device_count()) # 输出GPU名称（GPU版） if torch.cuda.is_available(): print("GPU名称：", torch.cuda.get_device_name(0))

成功结果：
- CPU 版：CUDA是否可用：False，但 PyTorch 版本正常输出，无报错；
- GPU 版：CUDA是否可用：True，并显示 GPU 数量和名称。

五、服务器环境常见问题解决

nvidia-smi 找不到命令：
- 原因：未安装 NVIDIA 驱动；
- 解决：联系服务器管理员安装驱动，或自行安装（需 root 权限）。
CUDA 版本不匹配报错：
- 原因：安装的 PyTorch CUDA 版本高于驱动支持的版本；
- 解决：降级 PyTorch 至匹配的 CUDA 版本（如驱动显示 CUDA 11.8，则安装 pytorch-cuda=11.8）。

conda 安装速度慢：

解决：添加国内镜像源（清华源）：

bash

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --set show_channel_urls yes