Loaded runtime CuDNN library: 5005 深入MNIST

来源:互联网 发布:国外数据采集 的方法 编辑:程序博客网 时间:2024/06/05 17:17

原文题:

tensorflow/stream_executor/cuda/cuda_dnn.cc:368] Loaded runtime CuDNN library: 5005 (compatibility version 5000) but source was compiled with 5103 (compatibility version 5100). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.

解决问题,就是更新cudnn

直接去cudnn官网上下载最新的cudnn.



下载cuDNN,从https://developer.nvidia.com/rdp/cudnn-download,注册,然后下载,

拷贝cuDNN库文件到cuda目录下,

tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgzcd cudasudo cp lib64/* /usr/local/cuda/lib64/sudo cp include/cudnn.h /usr/local/cuda/include/

设置环境变量,在/etc/profile中添加CUDA环境变量,

sudo gedit /etc/profile

在打开的文件中加入如下两句话

export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

保存后,使环境变量立即生效,

source /etc/profile

进入/usr/local/cuda/samples,执行下面的命令来build samples,

sudo make all -j4

全部编译完成后,进入 samples/bin/x86_64/linux/release,运行deviceQuery,

./deviceQuery

0 0
原创粉丝点击