【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

你的名字 2022-10-22 15:57 323阅读 0赞

# 1 问题 #

使用model.cuda(),作用是对于模型还是数据，cuda()函数都能实现从CPU到GPU的内存迁移

> model.cuda()

报错  
Exception has occurred: RuntimeError cuDNN error: CUDNN\_STATUS\_NOT\_INITIALIZED

# 2 原因与解决方案 #

## 2.1 方法一 ##

原因是：可能是Pytorch需要安装的四个包，版本没有对应，导致CUDA没法用

pytorch==1.7.1
    torchvision==0.8.2
    torchaudio==0.7.2
    cudatoolkit==10.*

卸载了所有与的四个包，按照官方提供的四个对应版本重新安装。当然首先需要查看自己的CUDA版本  
[Pytorch官网安装Pytorch][Pytorch_Pytorch]

> nvcc -V

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzkzNTY5Ng_size_16_color_FFFFFF_t_70]  
本人是Linux系统CUDA10，命令如下

#CUDA 10.1
    conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

## 2.2 方法二 ##

原因是：CUDA与Ｎ卡驱动不对应，没有安装正确，CUDA不能用，查看CUDA是否能用命令如下

import torch
    print(torch.cuda.is_available())

如果返回False，则重新安装CUDA，Linux系统具体参考本人另一篇博客  
[【Deepin 20系统】自定义路径安装CUDA-10或CUDA-11+cuDNN][Deepin 20_CUDA-10_CUDA-11_cuDNN]

## 2.3 方法三 ##

原因是：版本太高  
将torch的一系列依赖包降级到1.7.1

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

## 2.4 方法四 ##

GPU实在用不了，换CPU.用model.to(device)替代model.cuda()

device = "cpu"
    model = model.to(device)

[Pytorch_Pytorch]: https://pytorch.org/get-started/previous-versions/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzkzNTY5Ng_size_16_color_FFFFFF_t_70]: /images/20221022/ae45e7d48dbc4ced8d2ae102683f8c0a.png
[Deepin 20_CUDA-10_CUDA-11_cuDNN]: https://betterbench.blog.csdn.net/article/details/110133068