使用cuda训练时可能遇到,是调用CUDNN时发生的。
(使用CPU训练没有这个问题,
torch.backends.cudnn.enabled = False #关闭cudnn使用CPU进行训练(可能降低训练速度)
)
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input.
当输入模型的torch的长度大于65535时,可能出现次错误。可以对数据进行切分,减少torch的长度。
可能和硬件参数有关,但是我只有3070的显卡,可能朋友们也会遇到不同的限制,可以考虑torch长度方向的问题。
如果不能解决,可以参考以下三个方向进行尝试: