ubuntu 快速安装 pytorch

一、pytorch环境搭建

从官网pip安装速度太慢，这里介绍一种快速安装方法，首先打开在终端中输入

nvidia-smi

查看cuda版本，注意11.4的可以装11.1 11.2 不一定得是11.4；

其次去pytorch官网确定要安装的pytorch ，例如安装 ：

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

接着去pytorch whl源 ，下载要安装的torch、torchvision、torchaudio，注意版本要对应；

下载完成直接 pip install xxx/.whl；

最后检测pytorch安装成功

import torch
torch.cuda.is_available()
返回True则表示安装成功