cuDNN error: CUDNN"/>
解决RuntimeError: cuDNN error: CUDNN
报错:
解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
原因:
1.可能cuda与python版本不一致
详见:
2.可能是你的GPU显存被占用光了,再运行就出错了。
我的用cpu,代码就能跑了
device = torch.device("cuda")
改为
device = torch.device("cpu")
更多推荐
解决RuntimeError: cuDNN error: CUDNN
发布评论