TensorFlow GPU版本安装(2):cudnn安装

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010736419/article/details/78134336

一 官方安装文档

PREREQUISITES

   CUDA 7.0 and a GPU of compute capability 3.0 or higher are required.

ALL PLATFORMS

   Extract the cuDNN archive to a directory of your choice, referred tobelow as <installpath>.

   Then follow the platform-specific instructions as follows.

LINUX

扫描二维码关注公众号,回复: 3811639 查看本文章

   cd <installpath>

   export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH

   Add <installpath> to your build and link process by adding-I<installpath> to your compile

line and-L<installpath> -lcudnn to your link line.

二 安装方法:

直接解压拷贝就可以了:

tar xvzf cudnn-8.0-linux-x64-v5.0-ga.tgz

sudo cp cuda/include/cudnn.h/usr/local/cuda/include

sudo cp cuda/lib64/*.*/usr/local/cuda/lib64

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

猜你喜欢

转载自blog.csdn.net/u010736419/article/details/78134336