CUDNN安装

时间:2017-03-31 14:25:32

标签: tensorflow cudnn

我正在尝试安装CUDNN以在我的Macbook Pro上使用支持GPU的tensorflow。从NVDIA网站下载cudnn后,这些都是我能找到的指示:

OS X

cd <installpath>
export DYLD_LIBRARY_PATH=`pwd`:$DYLD_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.

虽然,第一步非常简单,第二个陈述的含义是什么,如何完成第二步? 请指教。感谢。

1 个答案:

答案 0 :(得分:2)

所以我被困在同一个地方,我发现cudnn存档基本上只有两个库文件:.h和.so文件。所以我认为最好把它们放在已经看过的地方,我发现了这个:

Untar the folder and copy the necessary files to the existing cuda 
library we installed earlier.

tar -xzvf cudnn-7.0-linux-x64-v4.0-prod.tgz
cp cuda/lib64/* /usr/local/cuda/lib64/
cp cuda/include/cudnn.h /usr/local/cuda/include/

所以我猜你应该找到mac的等价物

来源:

https://medium.com/@acrosson/installing-nvidia-cuda-cudnn-tensorflow-and-keras-69bbf33dce8a