RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED解决办法

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 问题可能有两种解决思路:

        1 检查GPU服务器的CUDA版本,一定要在Pytorch官网安装对应你CUDA版本的Pytorch

        2 若提示loss不能反传,则检查分类类别数量是否与实际数据的类别数量一致。

       

你可能感兴趣的:(问题处理记录,深度学习,pytorch,人工智能)