pytorch使用GPU时,运行缓慢,请问是驱动的原因还是其他原因?

比如说,执行c=torch.rand(5).cuda()时,要等很久,才执行完。这是显卡驱动程序装的版本不合适吗,还是其他原因?用的是1080的显卡
已邀请:

Spencer

赞同来自:

你好 请问你问题解决了吗? 我使用GPU时候一直没有响应,看GPU状态温度和占用显存一直在升高。

要回复问题请先登录注册