Ubuntu18.04安装cudnn v8

本着不折腾难受的原则,装一下cudnn
首先,下载安装包
下载 cuDNN Libaray for Linux 和下面的三个 Runtime、Developer 和 Code Samples
在这里插入图片描述
然后开始安装

  1. 解压 cuDNN 代码
tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
  1. 将文件复制到 CUDA 路径下(这个路径我并没有直接设置为cuda,好像是默认的cuda-xx.x这样一个文件名,请按自己实际情况自行修改)
sudo cp cuda/include/cudnn*.h /usr/local/cuda-xx.x/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-xx.x/lib64
sudo chmod a+r /usr/local/cuda-xx.x/include/cudnn*.h /usr/local/cuda-xx.x/lib64/libcudnn*
  1. cd到三个包的下载目录下,然后按照顺序运行:

安装 runtime library:

sudo dpkg -i libcudnn8_x.x.x-1+cudax.x_amd64.deb

安装developer library:

sudo dpkg -i libcudnn8-dev_8.x.x.x-1+cudax.x_amd64.deb

安装 code samples 和 cuDNN library documentation:

sudo dpkg -i libcudnn8-doc_8.x.x.x-1+cudax.x_amd64.deb
  1. 使用例子测试是否安装成功

复制 cuDNN sample到有读写权限的文件夹下:

$ cp -r /usr/src/cudnn_samples_v8/ $HOME

cd到复制的mnistcudnn目录下

$ cd  $HOME/cudnn_samples_v8/mnistCUDNN

编译mnistCUDNN sample.

$make clean && make

运行mnistCUDNN sample.

$ ./mnistCUDNN

一番运行后,出现如下信息表明安装成功:

Test passed!

参考链接:

  1. https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#cross-compiling-cudnn-overview
  2. https://zhuanlan.zhihu.com/p/68544360
  • 8
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
共两个压缩包(解压后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: 。。。。。
在Ubuntu上安装cuDNN通常需要以下步骤: 1. **下载cuDNN**:首先访问NVIDIA官网(https://developer.nvidia.com/cudnn),下载适合你的CUDA版本的cuDNN。注意选择适用于Ubuntu的版本,并根据你的GPU架构(比如CPU系列)选择合适的包。 2. **更新依赖**:在安装之前,确保你的系统已经是最新的,可以使用`sudo apt update && sudo apt upgrade`命令。 3. **安装依赖**:cuDNN有时需要特定的依赖项才能安装,比如build-essential、cuda-cubin-tools等,可以使用`sudo apt install build-essential cuda-command-line-tools libncurses5-dev libcurl4-openssl-dev`。 4. **解压和配置**:将下载的cuDNN压缩包解压到合适的位置,然后创建一个名为`install.txt`的文件,内容类似`-- Silence is golden - /usr/local/cuda`,这告诉cuDNN安装在哪里。 5. **运行安装脚本**:进入解压目录,执行`bash install_file_name.run`,这里填写你刚才创建的安装文件名。 6. **完成安装**:脚本会引导你完成剩余的安装过程,包括添加环境变量和设置库路径。 7. **验证安装**:安装完成后,你可以按照前面提到的方法检查cuDNN安装位置、环境变量设置以及运行测试程序来验证安装是否成功。 8. **添加开机启动**:为了每次启动都无需手动配置,可以将cuDNN的相关环境变量添加到`~/.bashrc`或`~/.profile`文件中,然后运行`source ~/.bashrc`(或重启)使更改生效。 记住,不同的cuDNN版本可能会有不同的安装步骤,确保遵循官方文档中的指导。
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值