python - 为什么我的 gpu 没有在 torch 中使用？

Question

我一直在视频数据集上训练模型，这是我的训练

我已经为模型、损失和加载器初始化了 Cuda。一切运行良好，这里是 nvidia-smi，它也在使用内存

但是它没有使用GPU，请在下面的任务管理器中查看，

谁能告诉我为什么会这样？我保留了影响训练准确性和运行时间的 6 批大小，因此，我无法对模型进行更改，因为它需要尝试。

标签： pythonpytorchgpu