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



所有评论(0)