Could not load dynamic library ‘libcudnn.so.8‘;

在尝试过许多办法之后,最终找到解决办法:

1.找到服务器的cuda版本,实验室服务器操作环境为linux/ubuntu20.04,cuda11.4(可在/usr/local/路径下查看)

2.找到合适cuda11.4的tensorflow-gpu版本,新建环境或者卸载所有之前和tensorflow相关的依赖包,用conda而不是pip安装,这里安装了2.1.0版本

~$ conda install tensorflow-gpu==2.1.0

 3.设置环境变量LD_LIBRARY_PATH

~$ vi ~/.bashrc

 按a进入编辑,加入以下代码

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu
export PATH=/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH

 ecs退出,:wq保存,最后使用如下命令对.bashrc文件的修改进行激活生效

~$ source ~/.bashrc

4.修改完测试一下

import os
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
gpu_device_name = tf.test.gpu_device_name()
print(gpu_device_name)

gpus = tf.config.list_physical_devices('GPU')
cpus = tf.config.list_physical_devices('CPU')
print(gpus, cpus)

看到运行结果,GPU可以成功被调用就可以了

/device:GPU:0
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU'), PhysicalDevice(name='/physical_device:GPU:1', device_type='GPU'), PhysicalDevice(name='/physical_device:GPU:2', device_type='GPU'), PhysicalDevice(name='/physical_device:GPU:3', device_type='GPU'), PhysicalDevice(name='/physical_device:GPU:4', device_type='GPU')] [PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]

Process finished with exit code 0

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值