gpu占用率一直不到百分之1
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
输出
True
1
然后是一些配置。win10、笔记本1650ti,版本都是按官方推荐下载的
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)
哪里出问题的
显存占用多少?不要看显卡使用率,这个是平均使用率,如果模型小数据少,使用率低很正常