pytorch安装

安装分两步

1.到官网，根据自己的系统找到想要的版本，如此处的v1.4.0

2.安装对应版本的wheel（以此处的1.4.0为例）

3.检查是否装好，不报错就好了

python #进入python环境输入以下命令 import torch import torchvision

4.补充

sudo lshw -c video #显示显卡信息 nvcc -V #显示终端信息 python --version #查看python版本