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

Question

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

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

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

然后也给模型、损失、数据指定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)

哪里出问题的

Answer 1

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