linux上tensorflow遇到Could not load dynamic library ‘libcudart.so.10.1‘解决方案

记录tensorflow无法使用GPU解决方案

问题描述

>>> import tensorflow
2021-04-24 11:57:36.192957: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.2/lib64:/usr/local/cuda-10.2/extras/CUPTI/lib64:
2021-04-24 11:57:36.192996: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

原因

没有正确配置好cuda

解决方案

1.官网查看tensorflow、python和cuDNN、CUDA的版本1

链接: link

2.利用清华源安装对应的cudnn版本、cudatoolkit版本
我的需求是这样

版本Python 版本cuDNNCUDA
tensorflow-2.3.03.5-3.87.610.1
conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
conda install cudnn=7.6 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

清华源的路径写的好像有点问题,不过还是成功下载了,没有报错等信息

3.测试

>>> import tensorflow
2021-04-24 12:18:40.298589: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1

运行任务也没有报错之类,用GPU详情命令2

nvidia-smi

可以看到GPU信息,并且Process那里可以看到你运行任务的PID
在这里插入图片描述
[1]:https://blog.csdn.net/fr555wlj/article/details/109155791
[2]:https://blog.csdn.net/sunjinshengli/article/details/108181660


  1. 参考 conda+清华园镜像安装指定版本的cudnn 链接: link ↩︎

  2. 参考 nvidia CUDA 信息查看 链接:link ↩︎

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值