cuDNN安装

xiaoxiao2021-02-28  73

最近在用tensorflow跑程序,要用到cuDNN,顺便弥补下以前用caffe没安装cuDNN的偷懒

随手记录下安装方法

安装cuDNN

Theano也支持cuDNN(可选),而Tensorflow则必需要cuDNN

首先去cuDNN官网下载文件,官网下载cuDNN v5.1 Library for Linux即可,解压缩后里面其实就运行库。

cuDNN download(需要注册),下载完毕后执行 

解压缩 >tar xvzf cudnn-7.0-Linux-x64-v4.0-prod.tgz 

复制移动文件 >sudo cp cuda/include/cudnn.h /usr/local/cuda/include  >sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 

创建新的软连接 >sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

即.h文件和运行库的软连接:

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

转载请注明原文地址: https://www.6miu.com/read-72721.html

最新回复(0)