在使用pytorch编译神经网络的时候报错
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
记录下我报错的原因:
1. 网络最后的输出数对应不上。(比方说,模型做的是4分类,但网络输出数是3)
2. 输入数据的label错误。(举例,如果做的是3分类,label应该是0,1,2;如果label写成-1,0,1,就会报错)
其他的报错原因可考虑:
1. batchsize不适合(gpu算力不足?)
2. pytorch和cudn版本不适配
3. 输入数据没有放到同一设备(gpu或cpu)上