Linux系统PyTorch安装步骤是什么

以下是在Linux系统上安装PyTorch的步骤：

安装Python和pip

大多数Linux发行版默认安装了Python，可使用以下命令检查：

python3 --version
pip3 --version

若未安装，对于Debian系（如Ubuntu），可执行：

sudo apt update
sudo apt install python3 python3-pip

对于RPM系（如CentOS），可执行：

sudo yum install python3 python3-pip

创建虚拟环境（可选）

为避免依赖冲突，建议创建虚拟环境：

python3 -m venv pytorch-env
source pytorch-env/bin/activate

安装PyTorch

• 使用pip安装：
  • CPU版本：pip install torch torchvision torchaudio
  • GPU版本（需先安装CUDA和cuDNN）：pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118（将cu118替换为实际CUDA版本号）
• 使用conda安装：
  • 安装Miniconda（若未安装）：

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

- **创建并激活环境**：

conda create -n pytorch-env python=3.8
conda activate pytorch-env

- **安装PyTorch**：
    - **CPU版本**：`conda install pytorch torchvision torchaudio cpuonly -c pytorch`
    - **GPU版本**：`conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia`（将`11.8`替换为实际CUDA版本号）

验证安装

在Python环境中运行以下代码：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 若为GPU版本，应返回True

配置环境变量（可选）

若使用GPU，需配置CUDA路径，在~/.bashrc或~/.zshrc中添加：

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后运行：

source ~/.bashrc  # 或 source ~/.zshrc