导入 tensorflow 版本 1.0 gpu 时出错

导入 tensorflow 版本 1.0 gpu 时出错

错误消息:我有带有 CUDA 8.0 和 cuDNN 5.1 的 ubuntu 16.04

Python 2.7.12(默认,2017 年 11 月 20 日,18:23:56)[GCC 5.4.0 20160609] 在 linux2 上输入“help”、“copyright”、“credits”或“license”以获取更多信息。

将 tensorflow 导入为 tf I tensorflow/stream_executor/dso_loader.cc:135] 成功在本地打开 CUDA 库 libcublas.so.8.0 I tensorflow/stream_executor/dso_loader.cc:126] 无法打开 CUDA 库 libcudnn.so.5。LD_LIBRARY_PATH: I tensorflow/stream_executor/cuda/cuda_dnn.cc:3517] 无法加载 cuDNN DSO I tensorflow/stream_executor/dso_loader.cc:135] 成功在本地打开 CUDA 库 libcufft.so.8.0 I tensorflow/stream_executor/dso_loader.cc:126] 无法打开 CUDA 库 libcuda.so.1。 LD_LIBRARY_PATH:I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:165] 主机名:harsh-Inspiron-3542 I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:189] libcuda 报告的版本为:未找到:无法找到 libcuda.so DSO 加载到此程序中 I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:363] 驱动程序版本文件内容:“”“NVRM 版本:NVIDIA UNIX x86_64 内核模块 340.104 Thu Sep 14 17:13:13 PDT 2017 GCC 版本:gcc 版本 5.4.0 20160609(Ubuntu 5.4.0-6ubuntu1~16.04.5)“”“I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:193] 内核报告的版本是:340.104.0 I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:1065] LD_LIBRARY_PATH:I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:1066] 未能在此系统上找到 libcuda.so:失败的先决条件:无法 dlopen DSO:libcuda.so.1;dlerror:libcuda.so.1:无法打开共享对象文件:没有此文件或目录 I tensorflow/stream_executor/dso_loader.cc:135] 成功在本地打开了 CUDA 库 libcurand.so.8.0

答案1

可以在这里找到。sudo apt install libcuda1-384

为了完整性,该文件应位于 /usr/lib/x86_64-linux-gnu/libcuda.so.1。

https://github.com/tensorflow/tensorflow/issues/4267

相关内容