Linux环境PyTorch安装步骤是什么

Linux环境PyTorch安装步骤

1. 安装系统依赖项

在安装PyTorch前，需确保系统具备必要的编译和运行环境。根据Linux发行版选择对应命令：

• Debian/Ubuntu系统：

  sudo apt update && sudo apt upgrade -y
  sudo apt install -y build-essential cmake git libopenblas-dev liblapack-dev libjpeg-dev libpng-dev libtiff-dev libavcodec-dev libavformat-dev libswscale-dev python3-dev python3-pip
  

• RPM/CentOS系统：

  sudo yum groupinstall "Development Tools" -y
  sudo yum install -y gcc-c++ cmake git openblas-devel lapack-devel libjpeg-turbo-devel libpng-devel libtiff-devel ffmpeg-devel python3-devel python3-pip
  

  以上命令安装了Python开发头文件、pip包管理器、CMake编译工具及线性代数库等依赖。

2. 安装Python与pip

多数Linux系统预装Python3，但需确认版本（建议Python≥3.8）及pip是否安装：

python3 --version  # 检查Python版本
pip3 --version     # 检查pip版本

若未安装，使用以下命令安装：

• Debian/Ubuntu：

  sudo apt install -y python3 python3-pip
  

• RPM/CentOS：

  sudo yum install -y python3 python3-pip
  

3. 创建虚拟环境（推荐）

为避免项目依赖冲突，建议使用虚拟环境隔离PyTorch及其依赖：

• 使用venv（Python内置工具）：

  python3 -m venv pytorch_env  # 创建名为pytorch_env的虚拟环境
  source pytorch_env/bin/activate  # 激活虚拟环境（激活后终端提示符会显示环境名）
  

• 使用conda（Anaconda/Miniconda用户）：

  conda create -n pytorch_env python=3.8 -y  # 创建名为pytorch_env的环境，指定Python版本
  conda activate pytorch_env  # 激活环境
  

4. 安装PyTorch

根据是否使用GPU加速，选择对应安装方式：

• CPU版本（无GPU加速）：
  直接通过pip安装PyTorch的CPU版本：

  pip3 install torch torchvision torchaudio
  

• GPU版本（需NVIDIA GPU+CUDA支持）：
  需提前安装NVIDIA驱动（版本≥450）和CUDA Toolkit（如11.8），然后通过pip安装对应版本的PyTorch：

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

  将cu118替换为实际安装的CUDA版本（如cu117、cu121等）。
• conda安装（推荐，更便捷）：
  若使用conda，可直接指定CUDA工具包版本安装：

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

  同样，将11.8替换为实际CUDA版本。

5. 验证安装

安装完成后，通过Python代码验证PyTorch是否安装成功及GPU是否可用：

python3 -c "import torch; print('PyTorch版本:', torch.__version__); print('CUDA可用:', torch.cuda.is_available())"

• 若输出PyTorch版本号（如2.1.0）且CUDA可用显示True，则说明GPU版本安装成功；
• 若CUDA可用显示False，则需检查CUDA驱动及环境变量配置。

6. 配置CUDA（GPU用户可选）

若使用GPU加速，需确保CUDA工具包正确安装并配置环境变量：

• 安装CUDA Toolkit：
  从NVIDIA官网下载对应版本的CUDA Toolkit（如11.8），运行安装脚本：

  wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
  sudo sh cuda_11.8.0_520.61.05_linux.run
  

  安装过程中选择“Continue”并接受许可协议，安装完成后重启系统。
• 配置环境变量：
  编辑~/.bashrc文件，添加以下内容：

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

  运行source ~/.bashrc使配置生效。

7. IDE配置（可选）

若使用PyCharm、VSCode等IDE，需将解释器设置为虚拟环境中的Python：

• PyCharm：进入File > Settings > Project > Python Interpreter，点击“+”添加解释器，选择虚拟环境中的python可执行文件（如pytorch_env/bin/python）。
• VSCode：按下Ctrl+Shift+P打开命令面板，输入“Python: Select Interpreter”，选择虚拟环境对应的Python路径。

以上步骤覆盖了Linux环境下PyTorch安装的全流程，根据自身需求选择CPU/GPU版本，并注意依赖项和环境变量的配置即可顺利完成安装。