qq_42923683 2021-05-13 18:05 采纳率: 100%
浏览 4444
已采纳

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

我在练习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进行训练?

  • 写回答

3条回答 默认 最新

  • CSDN专家-黄老师 2021-05-13 18:11
    关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?