您的位置:首页 > 其它

import TensorFlow提示Unable to load cuDNN DSO

2015-12-20 02:54 381 查看

import TensorFlow提示Unable to load cuDNN DSO

系统版本:Ubuntu 15.04

Python版本:2.7

cuda版本:7.0 (输入
nvcc --version
可查看版本号)

cuDNN版本: 7.0

错误原因:

目前TensorFlow不支持7.0版本的cuDNN,最高支持cuDNN 6.5 v2。

解决方法:

下载cuDNN 6.5 v2:

前往NVIDIA官网下载6.5版本的cuDNN。老版本可以在Archived cuDNN Releases选项中找到,下载cuDNN v2 Library for Linux而不是R1版本,文件名是cudnn-6.5-linux-x64-v2.tgz。

解压并安装:

说是安装实际上就是把解压后的文件复制到对应的includ和lib目录并链接。

复制cuDNN文件

进入解压后的文件夹,输入下面命令:

$ sudo cp cudnn.h /usr/local/include
$ sudo cp libcudnn.so /usr/local/lib
$ sudo cp libcudnn.so.6.5 /usr/local/lib
$ sudo cp libcudnn.so.6.5.48 /usr/local/lib


链接cuDNN的库文件

$ sudo ln -sf /usr/local/lib/libcudnn.so.6.5.48 /usr/local/lib/libcudnn.so.6.5
$ sudo ln -sf /usr/local/lib/libcudnn.so.6.5 /usr/local/lib/libcudnn.so
$ sudo ldconfig -v


吐槽一句,支持老版本不支持新版本,难不成真的是上一代技术拿出来放烟的吗,唉。。。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: