运行Tensorflow以使用我的GPU的问题

时间:2020-06-13 05:23:57

标签: python-3.x tensorflow

我已经从NVIDIA网站完成了cuda驱动程序以及cuDNN兼容版本的安装。

进一步,当我运行以下python代码

from tensorflow.python.client import device_lib print(device_lib.list_local_devices())

我遇到以下错误-

2020-06-12 23:14:13.148029:W tensorflow / stream_executor / platform / default / dso_loader.cc:55]无法加载动态库'libcudnn.so.7'; dlerror:libcudnn.so.7:无法打开共享对象文件:没有这样的文件或目录

我检查了cuda所在的目录,并位于lib64〜libcudnn.so.7下。我不确定在这里做什么。在检查其他查询时,我尝试更改某些目录的路径,但无济于事。

我正在运行具有cuda 10.1和cuDNN版本7.6.5的Ubuntu 20.04,该版本与tensorflow 2.2.0兼容

我还下载了tensorflow和tensorflow-gpu。

请帮助我找到相同的解决方案。

0 个答案:

没有答案