【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

简介: 本文提供了在使用PyTorch时遇到cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误的几种解决方法,包括重新安装PyTorch及其相关库以确保版本兼容,检查CUDA与显卡驱动的对应关系,以及在无法使用GPU的情况下切换到CPU运行模型。

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

nvcc -V

1.png


本人是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与N卡驱动不对应,没有安装正确,CUDA不能用,查看CUDA是否能用命令如下

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

如果返回False,则重新安装CUDA,Linux系统具体参考本人另一篇博客
【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)
目录
相关文章
|
1月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
3274 3
|
并行计算 Ubuntu PyTorch
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(下)
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(上)
309 0
|
并行计算 Ubuntu PyTorch
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(上)
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10
377 0
|
并行计算 Ubuntu PyTorch
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)(上)
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)
367 0
|
6月前
|
机器学习/深度学习 并行计算 算法框架/工具
Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置
Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置
888 3
|
并行计算 Ubuntu PyTorch
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)(下)
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)
216 0
|
并行计算 PyTorch 算法框架/工具
Pytorch安装和检验cuda和cudnn是否可用
Pytorch安装和检验cuda和cudnn是否可用
1545 0
Pytorch安装和检验cuda和cudnn是否可用
|
机器学习/深度学习 并行计算 PyTorch
【RTX 3060Ti 深度学习环境配置图文(安装Anaconda、VScode、CUDA、CUDNN、pytorch)】
【RTX 3060Ti 深度学习环境配置图文(安装Anaconda、VScode、CUDA、CUDNN、pytorch)】
399 0
【RTX 3060Ti 深度学习环境配置图文(安装Anaconda、VScode、CUDA、CUDNN、pytorch)】
|
机器学习/深度学习 并行计算 PyTorch
深度学习环境安装(anaconda、pycharm、cudn、cudnn、pytorch)
深度学习环境安装(anaconda、pycharm、cudn、cudnn、pytorch)
|
PyTorch 网络安全 算法框架/工具
下载Pytorch的自带数据集时报错=urllib.error.URLError: urlopen error [SSL: CERTIFICATE_VERIFY_FAILED]
这是一个SSL证书验证错误,当请求一个https站点,但是证书验证错误时,就会报这样的错误。
286 0

热门文章

最新文章