调用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:一个大佬通过国外一篇论文仿的,我以为不会出现这种小毛病自我怀疑好久,不该把不该把

你可能感兴趣的:(废掉小脑阔的错误,深度学习,pytorch,神经网络)