RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 问题可能有两种解决思路:
1 检查GPU服务器的CUDA版本,一定要在Pytorch官网安装对应你CUDA版本的Pytorch
2 若提示loss不能反传,则检查分类类别数量是否与实际数据的类别数量一致。
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 问题可能有两种解决思路:
1 检查GPU服务器的CUDA版本,一定要在Pytorch官网安装对应你CUDA版本的Pytorch
2 若提示loss不能反传,则检查分类类别数量是否与实际数据的类别数量一致。