pytorch 模型在GPU上但训练时仍使用CPU

Question

我在练习pytorch，使用这个页面的代码。https://www.pytorch123.com/SecondSection/training_a_classifier/

其中提到:如果使用GPU训练，代码需要更改

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)
...
...
inputs, labels = inputs.to(device), labels.to(device)

其余部分没有改动，但是运行时资源管理器显示GPU占用0%，并且CPU保持一个恒定的高占用率。是否说明我没有成功在GPU上训练？

请问怎么才能使用CPU进行训练？

Answer 1

你要安装cuda和cudnn，参考一下https://zhuanlan.zhihu.com/p/106133822