安装Pytorch+cuda10.2+cudnn（笔记本集显+独显）踩坑

项目场景：

安装pytorch+cuda+cudnn遇到的离谱事件

---

问题描述：

之前安tensorflow，能够检测到电脑的两张显卡，集显+独显，于是独显编号为1。由于先入为主，安装pytrorch我也有这样的想法。

torch.cuda.device_count()#返回值一直为1

一直以为是自己版本安装错误，于是安完cuda11又卸载又安10.0，再卸载，再安10.2
最后还是1。

---

解决方案：

安了10.2，想随便找个测试，监测独显使用率。
结果如下，感觉自己好蠢。一开始就该get_device_name的。
记录一下。

测试代码转载自：

https://blog.csdn.net/weixin_35576881/article/details/89709116`

---