首页 > 解决方案 > 使用 Cuda 和 Pytorch 使用 GPU

问题描述

当我开始将 GPU 用于 ML 应用程序时,我去任务管理器查看它使用了多少 GPU,并找到了六个不同的图表,其中只有两个显示活动。Compute_0复制Compute_0的间隔介于 85-95 % 和Copy 5-15 % 之间,但它仅代表 GPU 全部使用的 8-12%。

我想知道它到底是如何工作的,并找出是否有办法增加 8-12%,因为它也是另一个标题为Compute_1的图表,表示使用量为零,暗示(按名称)它也可以使用。

这是实现代码:

if torch.cuda.is_available():
    device = torch.device('cuda:0')

我正在使用 NVIDIA GeForce 940MX

标签: gpupytorch

解决方案


推荐阅读