Pytorch 模型将 GPU 设置为在 Nvidia gpu 上运行答案

【问题标题】：Pytorch Model set GPU to run on Nvidia gpuPytorch 模型将 GPU 设置为在 Nvidia gpu 上运行
【发布时间】：2021-06-19 14:48:20
【问题描述】：

我正在学习机器学习并尝试在我的 Nvidia GTX 1650 上运行模型（Pytorch）。

torch.cuda.is_available() => True
model.to(device)

实现了以上几行在GPU上运行模型，但任务管理器显示两个GPU

1. Intel Graphics
2. Nvidia GTX 1650

CPU 使用率的波动显示在 Intel 而不是 Nvidia 上。

如何在 Nvidia GPU 上运行它？

注意：代码运行良好，在 Intel 上执行，历时约 90-100 秒。

【问题讨论】：

【解决方案1】：

做吧

  device = torch.device("cuda:0")

【讨论】：

【解决方案2】：

试试这个，希望有用

device = 'cuda'
check = torch.cuda.current_device()
print(torch.cuda.get_device_name(check)) 

#model.to(device)

【讨论】：