一个帅气的boy,你可以叫我Love And Program
⌨个人主页:Love And Program的个人主页
如果对你有帮助的话希望三连支持一下博主
我在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:一个大佬通过国外一篇论文仿的,我以为不会出现这种小毛病自我怀疑好久,不该把不该把