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

2024-07-12 00:49| 来源: 网络整理| 查看: 265

项目场景：

安装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`

【本文地址】

今日新闻