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

Answer 2

看看你显卡驱动版本、cuda版本、cudnn版本、pytorch版本的对应情况。版本不对应也会出现都安装好了，还是使用cpu运行的。

Answer 3

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632