做图像分类时，发现应该是有GPU设备，但没有在GPU上运行，想问问是设备不行，还是代码有误。

不知道如何修正了，电脑xin shou，有无da lao帮忙看一下
做图像分类时，发现应该是有GPU设备，但没有在GPU上运行，想问问是设备不行，还是代码有误。

上图只能说明计算机有GPU，看起来应该是用 PyToch。但是：
（1）安装的 PyTorch 是 CPU版还是 GPU版本？
（2）程序中有没有将模型和数据加载到 GPU，并在 GPU 运行。
你都没有说。这是判断不出来问题的。