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