- 博客(2)
- 收藏
- 关注
RSS订阅原创 对上一章Attention_Seq2Seq_lstm模型运行时在CPU上的问题已经解决
在模型在GPU部署后，运行时在模型里创建的的变量会出现CPU上，直接把它们也部署在GPU。上就可以在GPU运行了（代码是B站和csdn上大佬们的代码合在一起的）
2024-06-08 17:14:09
272
原创 求大佬指点，模型和数据在GPU上但运行时CPU满载荷运行，GPU没反应，GPU是正常的，运行其他的模型没有问题
模型的部署在GPU，LOSS输出也在GPU，但是运算不在，还有其中必须要有logits=logits.to(device)才能运行，因为没有的话在第一次运行后模型输出的数据会在CPU上，模型的输入数据也在GPU上，但是运算时gpu利用率不到%2，cpu满载荷运行。
2024-06-05 14:24:20
353
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人