为啥始终不能用gpu跑pytorch框架

gpu占用率一直不到百分之1


import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())

输出
True
1
然后是一些配置。win10、笔记本1650ti,版本都是按官方推荐下载的

img

img


然后也给模型、损失、数据指定gpu了,部分代码如下

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
net = Net()
cost = torch.nn.MSELoss()  
net = net.to(device)  
cost = cost.to(device)  
for x, y in neural_train_data:
  x = x.to(device)
  y = y.to(device)

哪里出问题的

img

img

img

显存占用多少?不要看显卡使用率,这个是平均使用率,如果模型小数据少,使用率低很正常