linux环境下pytorch安装指南

在Linux环境下安装PyTorch可以分为几个主要步骤，具体取决于你是否需要使用GPU支持。以下是在Linux环境下安装PyTorch的详细指南：

使用Anaconda安装PyTorch（推荐）

1. 更新系统包

   sudo apt update && sudo apt upgrade -y
   

2. 安装Anaconda

   • 访问Anaconda官网，下载适用于Linux的Anaconda安装包。
   • 在终端中导航到下载目录，运行安装脚本：

     bash Anaconda3-2022.10-Linux-x86_64.sh
     

   • 按照屏幕上的指示完成安装，并在.bashrc文件中添加Anaconda的路径。

3. 创建虚拟环境

   conda create --name pytorch_env python=3.8
   conda activate pytorch_env
   

4. 安装PyTorch

   • 安装CPU版本：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • 安装GPU版本（确保已安装CUDA和cuDNN）：

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

5. 验证安装

   python -c "import torch; print(torch.__version__)"
   

使用pip安装PyTorch

1. 更新包管理器

   sudo apt-get update
   

2. 安装Python和基本库

   sudo apt-get install python3-pip
   pip3 install torch torchvision
   

3. 检查CUDA和cuDNN支持（如果使用GPU）

   • 确保你的系统已经安装了CUDA和cuDNN。
   • 安装PyTorch GPU版本：

     pip3 install torch torchvision -f https://download.pytorch.org/whl/cu118/torch_stable.html
     

4. 验证安装

   python -c "import torch; print(torch.__version__)"
   

请根据你的系统配置和需求选择合适的安装方法。如果你使用的是GPU版本，请确保你的系统已经正确安装了NVIDIA驱动和相应的CUDA版本。