cuDNN error: CUDNN_STATUS_BAD_PARAM

情况:loss.backward()中一直报错,数据类型没有问题。

这四块显卡跑就报错,换了个节点里的显卡 

就能运行 

你可能感兴趣的:(python)