pytorch官网中如何选择以及后面的安装和pycharm测试步骤

安装步骤

前面的安装步骤很容易搜到，这里就不赘述。

windows系统可以在任务管理器（设备中搜索）中看自己是否有GPU或者自己的GPU类型

如果你的显卡是NVIDIA，根据它的推荐选择CUDA版本，一般游戏本的配置会好一些，像我的是轻薄本配置就会不太适合。

像我的是AMD类型不支持CUDA，所以选择CPU，再运行下面的这串代码conda install pytorch torchvision torchaudio cpuonly -c pytorch

开始下载

下载完成显示done
以下代码可以测试，但注意我们下载的是cpu版本的，不是CUDA版本，所以后面两行代码是不能达到理想运行成果的。

import torch 

print(torch.__version__) # pytorch版本
print(torch.version.cuda) # cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用

在pycharm中搭载环境

选择file-----settings----python interpreter-----小齿轮-----add-----然后如下图所示

运行情况

from torchvision import models
print(dir(models))

可以运行