打开APP
userphoto
未登录

开通VIP,畅享免费电子书等14项超值服

开通VIP
import TensorFlow提示Unable to load cuDNN DSO

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
  • 1
  • 2
  • 3
  • 4
  • 1
  • 2
  • 3
  • 4

链接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
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3

吐槽一句,支持老版本不支持新版本,难不成真的是上一代技术拿出来放烟的吗,唉。。。

本站仅提供存储服务,所有内容均由用户发布,如发现有害或侵权内容,请点击举报
打开APP,阅读全文并永久保存 查看更多类似文章
猜你喜欢
类似文章
【Linux】在一台机器上同时安装多个版本的CUDA(切换CUDA版本)
手把手教你搭建计算机视觉开发环境
ubuntu系统问题
Ubuntu16.04+CUDA8.0+caffe配置
caffe+Ubuntu14.04.10 +cuda7.0/7.5+CuDNNv4 安装
Ubuntu16.04下nvidia驱动+cuda+cudnn+nccl安装
更多类似文章 >>
生活服务
热点新闻
分享 收藏 导长图 关注 下载文章
绑定账号成功
后续可登录账号畅享VIP特权!
如果VIP功能使用有故障,
可点击这里联系客服!

联系客服