Ubuntu20.04安装cuDNN8.0

在博客Tensorflow + PyTorch 安装(CPU + GPU 版本)中找到相应的版本匹配,确定好自己的电脑应该安装什么版本,再开始下载。

如果CUDA没有安装好的话要先安装CUDA

一、下载安装包

官网链接:cuDNN Download

这里需要登录一下,如果是新建账号的话可能需要填个问卷

进入链接后选择对应的版本和安装包。这里建议是使用for Linux而不是for Ubuntu ,因为Linux安装包是一个tar文件,之后可以解压

在这里插入图片描述

下载好以后进行安装

tar -xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz

二、文件复制

解压完成后生成一个cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive文件夹,运行如下命令

# 复制cudnn头文件
sudo cp cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/include/* /usr/local/cuda-11.7/include/  
# 复制cudnn的库
sudo cp cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/lib/* /usr/local/cuda-11.7/lib64/ 

三、测试

  • 给予权限
sudo chmod +x /usr/local/cuda-11.7/include/cudnn*.h
sudo chmod +x /usr/local/cuda-11.7/lib64/libcudnn*
  • 测试
cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

image-20220720221110394

说明已经成功安装

参考链接

ubuntu安装cudnn

Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)

最新CUDA环境配置教程(ubuntu 20.04 + cuda 11.7 + cuDNN 8.4)

ubuntu20.04 nvida驱动+cudu11.4+cudnn8.4.0安装

共两个压缩包(解压后145M),此为其一 环境: windows10(X64) vs2015 Anaconda3-4.4.0-Windows-x86_64(python3.6) cuda_8.0.61_win10(basic) cuda_8.0.61.2_windows(Patch 2) cudnn-8.0 tensorflow1.30 pip install tensorflow-gpu后 Anaconda prompt中输入:import tensorflow时,报错:ImportError,需要用到这个库文件,分别copy至CUDA\v8.0 中三个目录下:bin,include,lib 具体错误一般如下: File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 18, in swig_import_helper return importlib.import_module(mname) File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 986, in _gcd_import File "", line 969, in _find_and_load File "", line 958, in _find_and_load_unlocked File "", line 666, in _load_unlocked File "", line 577, in module_from_spec File "", line 906, in create_module File "", line 222, in _call_with_frames_removed ImportError: DLL load failed: The specified module could not be found. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "C:\Anaconda3\lib\site-packages\tensorflow\python\__init__.py", line 66, in from tensorflow.python import pywrap_tensorflow File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 21, in _pywrap_tensorflow = swig_import_helper() File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 20, in swig_import_helper return importlib.import_module('_pywrap_tensorflow') File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) ImportError: No module named '_pywrap_tensorflow' During handling of the above exception, another exception occurred: 。。。。。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值