pytorch Model.cuda() 执行的时间过长

即使对于一个小模型,调用cuda()方法也需要几分钟的时间。这个是正常的吗?如果是,那么为什么会花那么长时间呢?
已邀请:
匿名用户

匿名用户

赞同来自:

我们知道,我们一直在发送的二进制包存在问题,这只出现在某些GPU架构上。你能告诉一下你的GPU吗?你也可以考虑重装GPU
匿名用户

匿名用户

赞同来自:

我在GPU服务器运行也出问题了。我使用pip在whl上安装的Titan x, torch-0.1.11.post5-cp27,cuda8.0。
但是当我卸载它以后,再安装torch - 0.1.10。post2,它工作得很好。

要回复问题请先登录注册