pytorch使用多张显卡报错,求改证

同一段代码使用单显卡操作时没有任何问题,使用多显卡时则会出现

RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

错误,不知道这个错误是如何产生的,网上查询得出的最有可能的解答是数据只要不是float32就会产生这样的问题,但不做任何修改仅使用单显卡这样的问题就不会存在

实在是想不通了,求大神解答。

代码来源:github源代码

1个回答

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问