Tensorflow 缺少 libcusolver.so.10 和 libcudnn.so.8 两个库的解决办法

简介: Tensorflow 缺少 libcusolver.so.10 和 libcudnn.so.8 两个库的解决办法

0. 环境展示

Ubuntu系统 Ubuntu 20.04 focal

Tensorflow 2.4.1

Conda conda 4.8.3

Python Python 3.8.3

Cuda Cuda compilation tools, release 11.1, V11.1.105

cudnn cuDNN v8.1.0 (January 26th, 2021), for CUDA 11.0,11.1 and 11.2下载链接 点此跳转

1. libcusolver.so.10 动态库

我们可以发现在 cuda 的 lib64包下 有这么一个文件

但是 Tensorflow 提示找不到 libcusolver.so.10 库,莫非是版本问题,于是试了一下把 文件中的 11 的库复制了一份并改名为 libcusolver.so.10, 或者说就是重新给 libcusolver.so.11.0.1.105 创建了一个链接文件

发现这个库的报错没了。

(虽然解决了问题,但总感觉会出事儿,至少目前还没事,希望以后也没事)


2. libcudnn.so.8 动态库

看名字估计能才出来是因为缺少 cudnn 的库,于是我们去官网下载对应版本。官方地址: https://developer.nvidia.com/rdp/cudnn-archive#a-collapse804-111

我的 cuda 是 11.1 版本,所以下载第一个,下载好后就是一个压缩包,解压缩后就是一个名为 cuda 的文件夹

打开 cuda 文件夹发现里边有 include 和 lib64 两个文件夹,分别存放着头文件和动态库,其中我们需要的 libcudnn.so.8 动态库就在lib64里,可以查找一下。

将 include 里的文件复制到 你安装的 cuda 的include 里边,lib64 里的动态库也移动到你的 系统 cuda 的 lib64文件夹中即可

cp -r ./cuda/include/* /usr/local/cuda-11.1/include/
cp -r ./cuda/lib64/* /usr/local/cuda-11.1/lib64/

打开python 检查tensorflow

import tensorflow as tf
tf.test.is_gpu_available()

可以看到 动态库加载无报错,gpu资源也为True 可用

gpu测试样例

with tf.device('/gpu:0'):
  gpu_a = tf.random.normal([1, 100])
  gpu_b = tf.random.normal([100, 1])
  print(gpu_a.device, gpu_b.device)

gpu 没问题。

相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
相关文章
|
5月前
|
机器学习/深度学习 PyTorch TensorFlow
在深度学习中,数据增强是一种常用的技术,用于通过增加训练数据的多样性来提高模型的泛化能力。`albumentations`是一个强大的Python库,用于图像增强,支持多种图像变换操作,并且可以与深度学习框架(如PyTorch、TensorFlow等)无缝集成。
在深度学习中,数据增强是一种常用的技术,用于通过增加训练数据的多样性来提高模型的泛化能力。`albumentations`是一个强大的Python库,用于图像增强,支持多种图像变换操作,并且可以与深度学习框架(如PyTorch、TensorFlow等)无缝集成。
|
7月前
|
机器学习/深度学习 TensorFlow API
TensorFlow的扩展库:TensorFlow Probability与TensorFlow Quantum
【4月更文挑战第17天】TensorFlow的扩展库TensorFlow Probability和TensorFlow Quantum开辟了机器学习和量子计算新纪元。TensorFlow Probability专注于概率推理和统计分析,集成深度学习,支持贝叶斯推断和变分推断,提供自动微分及丰富的概率模型工具。其Bijector组件允许复杂随机变量转换,增强建模能力。另一方面,TensorFlow Quantum结合量子计算与深度学习,处理量子数据,构建量子-经典混合模型,应用于化学模拟、量子控制等领域,内置量子计算基元和高性能模拟器。
|
7月前
|
并行计算 TensorFlow 算法框架/工具
Linux Ubuntu配置CPU与GPU版本tensorflow库的方法
Linux Ubuntu配置CPU与GPU版本tensorflow库的方法
160 1
|
7月前
|
并行计算 TensorFlow 算法框架/工具
新版本GPU加速的tensorflow库的配置方法
新版本GPU加速的tensorflow库的配置方法
182 1
|
7月前
|
机器学习/深度学习 TensorFlow 算法框架/工具
Anaconda配置Python新版本tensorflow库(CPU、GPU通用)的方法
Anaconda配置Python新版本tensorflow库(CPU、GPU通用)的方法
157 1
|
人工智能 并行计算 Linux
Win10系统编译Tensorflow Lite 2.3为动态链接库tensorflowlite_c.dll
Win10系统编译Tensorflow Lite 2.3为动态链接库tensorflowlite_c.dll
Win10系统编译Tensorflow Lite 2.3为动态链接库tensorflowlite_c.dll
|
搜索推荐 TensorFlow 算法框架/工具
基于tensorflow和flask的本地图片库web图片搜索引擎
基于tensorflow和flask的本地图片库web图片搜索引擎
146 0
基于tensorflow和flask的本地图片库web图片搜索引擎
|
机器学习/深度学习 存储 人工智能
人工智能不过尔尔,基于Python3深度学习库Keras/TensorFlow打造属于自己的聊天机器人(ChatRobot)
聊天机器人(ChatRobot)的概念我们并不陌生,也许你曾经在百无聊赖之下和Siri打情骂俏过,亦或是闲暇之余与小爱同学谈笑风生,无论如何,我们都得承认,人工智能已经深入了我们的生活。目前市面上提供三方api的机器人不胜枚举:微软小冰、图灵机器人、腾讯闲聊、青云客机器人等等,只要我们想,就随时可以在app端或者web应用上进行接入。但是,这些应用的底层到底如何实现的?在没有网络接入的情况下,我们能不能像美剧[《西部世界》(Westworld)](https://movie.douban.com/subject/2338055/)里面描绘的那样,机器人只需要存储在本地的“心智球”就可以和人类沟
人工智能不过尔尔,基于Python3深度学习库Keras/TensorFlow打造属于自己的聊天机器人(ChatRobot)
|
机器学习/深度学习 存储 人工智能
TensorFlow?PyTorch?Paddle?AI工具库生态之争:ONNX将一统天下
AI诸多工具库工具库之间的切换,是一件耗时耗力的麻烦事。ONNX 即应运而生,使不同人工智能框架(如PyTorch、TensorRT、MXNet)可以采用相同格式存储模
2285 1
TensorFlow?PyTorch?Paddle?AI工具库生态之争:ONNX将一统天下
|
TensorFlow 算法框架/工具 Python
DeepCTR反复安装各种库numpy,tensorflow
DeepCTR反复安装各种库numpy,tensorflow
183 0