在Ubuntu下安装PyTorch（GPU版）的简易教程

准备工作

• 1.需要确保你的电脑上安装了CUDA，并且CUDA的版本支持PyTorch，可以参考PyTorch官网上提供的CUDA版本对照表来查看。
• 2.你需要确定你的电脑上是否安装了PyTorch的依赖库，比如说C++编译器、Python环境等，如果没有安装，可以参考PyTorch官网上提供的安装指南。

安装PyTorch

• 1.打开终端，输入以下命令来安装PyTorch：

  $ pip install torch torchvision
  

• 2.根据你的CUDA版本，输入以下命令来安装PyTorch：

  # 如果CUDA的版本是9.0
  $ pip install torch torchvision cuda90
  
  # 如果CUDA的版本是9.2
  $ pip install torch torchvision cuda92
  
  # 如果CUDA的版本是10.0
  $ pip install torch torchvision cuda100
  
  # 如果CUDA的版本是10.1
  $ pip install torch torchvision cuda101
  
  # 如果CUDA的版本是10.2
  $ pip install torch torchvision cuda102
  
  # 如果CUDA的版本是11.0
  $ pip install torch torchvision cuda110
  

• 3.输入以下命令来检查是否安装成功：

  $ python
  >>> import torch
  >>> print(torch.__version__)
  

  如果能够正确输出PyTorch的版本号，表明安装成功。