👦👦一个帅气的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 <class 'torch.device'>

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

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