win10【环境】pytorch（gpu版本）安装

感觉自己踩了好多雷。搞深度学习不易啊。以下是我安装的流程，希望能帮到大家，一起冲冲冲鸭。

---

设备信息：Gefoece RTX 2080 super
已安装：anaconda3和pycharm
安装参考链接：https://www.cnblogs.com/yuxuefeng/p/9235431.html

看了好多博客，都是先安装cuda,cudann，然后才装pytorch，开始，我是按着这个来的，为了安装cuda，我安装了visual studio 2013，安装了 NVIDIA 显卡驱动程序，装了对应版本的cuda11。。。。然鹅，打开pytorch我发现！人家才支持到cuda10.2，而且，也没有对应版本的cudnn。这让安装了一天的我，雪上加霜。吐血过后，还得继续整呀。
我看了少数人讲到安装pytorch时，它会自动的帮你安装你需要的（只要有 NVIDIA 显卡驱动程序)，这次我干脆把cuda11卸了。。。从头来过。

NVIDIA 显卡驱动程序安装

如果有，就不用安装了（一般刚装完系统都会安装这些驱动）
下载地址：http://www.nvidia.cn/Download/index.aspx?lang=cn

选择适合自己电脑的显卡驱动下载
安装很简单，直接下一步就可以
默认安装路径： C:\NVIDIA\DisplayDriver\390.77\Win8_Win7_64\International

pytorch安装

直接去官网下载https://pytorch.org/get-started/locally/

直接复制命令到命令窗口运行。

最后

如何检查pytorch是否正在使用GPU？
原文链接：https://blog.csdn.net/qq_35435964/article/details/100928501

import torch
a = torch.cuda.is_available()
print(a)
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

结果：
True
cuda:0
GeForce GTX 2080 SUPER
tensor([[0.9530, 0.4746, 0.9819],
[0.7192, 0.9427, 0.6768],
[0.8594, 0.9490, 0.6551]], device=‘cuda:0’)