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
共两个压缩包(解压后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: 。。。。。
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值