项目场景:

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

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐