Linux PyTorch安装步骤是什么

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

使用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-cuda11.8 -c pytorch -c nvidia
     

5. 验证安装：

   • 在Python环境中输入：

     import torch; print(torch.__version__)
     

   • 检查CUDA和cuDNN支持（如果使用GPU）：

     import torch
     print(torch.cuda.device_count())  # 查看可用的GPU个数
     print(torch.cuda.is_available())  # 检查CUDA是否可用
     

使用pip安装PyTorch

1. 更新包管理器：

   sudo apt update
   

2. 安装Python和基本库：

   sudo apt install -y python3 python3-pip
   

3. 创建虚拟环境（可选但推荐）：

   python3 -m venv pytorch_env
   source pytorch_env/bin/activate
   

4. 安装PyTorch：

   • CPU版本：

     pip3 install torch torchvision torchaudio
     

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

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

5. 验证安装：

   • 在Python环境中输入：

     import torch; print(torch.__version__)
     

   • 检查CUDA和cuDNN支持（如果使用GPU）：

     import torch
     print(torch.cuda.device_count())  # 查看可用的GPU个数
     print(torch.cuda.is_available())  # 检查CUDA是否可用
     

通过以上步骤，你应该能够在Linux系统下成功安装并运行PyTorch。如果遇到任何问题，可以参考PyTorch官方文档获取更多信息。