运行时cudnn报如下错误:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
解决办法:
1.先尝试如下:
import torch
torch.backends.cudnn.benchmark=True
2.查看torch cuda cudnn版本是否兼容
先尝试1,不要一上来就是卸载,以免越来越乱
运行时cudnn报如下错误:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
解决办法:
1.先尝试如下:
import torch
torch.backends.cudnn.benchmark=True
2.查看torch cuda cudnn版本是否兼容
先尝试1,不要一上来就是卸载,以免越来越乱