栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

调用GPU不匹配“cuda:0“

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

调用GPU不匹配“cuda:0“

一个帅气的boy,你可以叫我Love And Program
 ⌨个人主页:Love And Program的个人主页
如果对你有帮助的话希望三连支持一下博主

GPU不匹配"cuda:0"
  • 前言
  • 代码

前言

       我在Github上发现这一组代码,平平无奇,就是调用GPU,但是我运行的时候一直显示调用CPU,我看着明明是GPU啊。

代码
import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
if device == 'cuda:0':
    print('Using GPU : ')
    print(torch.cuda.get_device_name(device))
else :
    print('Using CPU')
# 结果
'''
CPU
'''

我尝试打印device,结果却是cuda:0。
然后我发现结果并不是想的那样

print(device,type(device))
#cuda:0 

很明显,类型都不一样,白瞎我一小时间

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)
if str(device) == 'cuda:0':
    print('Using GPU : ')
    print(torch.cuda.get_device_name(device))
else :
    print('Using CPU')
'''
cuda:0
Using GPU : 
GeForce MX130
'''
ps:一个大佬通过国外一篇论文仿的,我以为不会出现这种小毛病自我怀疑好久,不该把不该把
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/869158.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号