SegNet-pytorch icon indicating copy to clipboard operation
SegNet-pytorch copied to clipboard

pytorch版本

Open lwbmowgli opened this issue 3 years ago • 1 comments

我使用pytorch1.9版本出现了初始化错误:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED。 请问您的torch版本是?

lwbmowgli avatar Nov 08 '21 02:11 lwbmowgli

时间有点久远记不太清了,不过pytorch版本过高的确是会出现报错,可以试着降低版本,1.4或是1.2.

chen-zhoujian avatar Nov 08 '21 14:11 chen-zhoujian