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

来源:互联网 发布:淘宝怎么搜片子 编辑:程序博客网 时间:2024/05/16 05:59

一 官方安装文档

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

 

   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*

阅读全文
0 0