import TensorFlow提示Unable to load cuDNN DSO

本文详细介绍了在Ubuntu15.04环境下,当使用Python版本2.7和CUDA版本7.0时,遇到的无法加载cuDNN7.0的错误问题。提供了下载cuDNN6.5v2版本的步骤,并指导如何正确安装以解决该问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值