ubuntu 20.04系统安装pytorch

1.1 安装gcc

安装cuda之前，首先应该安装gcc，安装cuda需要用到gcc，否则报错。可以先使用下方指令在终端查看是否已经安装gcc。

gcc --version

如果终端打印如下则说明已经安装。

如果显示“找不到命令 “gcc”......”使用下方指令安装

sudo apt-get update
sudo apt-get install gcc

1.2 查看cuda版本      

step1：查看显卡驱动支持cuda最高版本

安装cuda之前首先需要查看系统显卡驱动所支持的最高cuda版本，在终端使用如下之指令就可以查看。

nvidia-smi

step2 :查看pytorch支持的cuda版本

本次安装pytorch，是直接用pip安装，不安装anconda。在安装之前，要确定安装的版本，比如cuda版本，去PyTorch官网上查看cuda版本，避免安装错误的版本。如下Ubuntu20.04对应的cuda版本为11.8：

1.3 cuda安装

Step 1: 下载cuda 

      去