在ubuntu18.04

时间:2018-11-27 09:52:06

标签: python tensorflow nvidia ubuntu-18.04

导入tensorflow时出现一些错误。

我的计算机的规格:

OS:ubuntu 18.04

Nvidia RTX 2080 Ti * 2

Nvidia驱动程序415

CUDA:10.0

cuDNN:7.3.0 张量流:1.11.0

import tensorflow

错误:

  

回溯(最近通话最近):文件   “ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py”,   第58行,在       从tensorflow.python.pywrap_tensorflow_internal导入*文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py”,   第28行,在       _pywrap_tensorflow_internal = swig_import_helper()文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py”,   第24行,在swig_import_helper中       _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路径名,描述)文件“ /usr/lib/python3.6/imp.py”,第243行,   在load_module中       在load_dynamic中返回load_dynamic(名称,文件名,文件)文件“ /usr/lib/python3.6/imp.py”,第343行       返回_load(spec)ImportError:libcublas.so.9.0:无法打开共享对象文件:没有这样的文件或目录

     

在处理上述异常期间,发生了另一个异常:

     

回溯(最近一次通话最后一次):文件“”,第1行,在      文件   “ /usr/local/lib/python3.6/dist-packages/tensorflow/init.py”,行   22,在       从tensorflow.python import pywrap_tensorflow#pylint:disable =未使用的导入文件   “ /usr/local/lib/python3.6/dist-packages/tensorflow/python/init.py”,   第49行,在       从tensorflow.python导入pywrap_tensorflow文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py”,   74行,在       引发ImportError(msg)ImportError:追溯(最近一次调用):文件   “ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py”,   第58行,在       从tensorflow.python.pywrap_tensorflow_internal导入*文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py”,   第28行,在       _pywrap_tensorflow_internal = swig_import_helper()文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py”,   第24行,在swig_import_helper中       _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路径名,描述)文件“ /usr/lib/python3.6/imp.py”,第243行,   在load_module中       在load_dynamic中返回load_dynamic(名称,文件名,文件)文件“ /usr/lib/python3.6/imp.py”,第343行       返回_load(spec)ImportError:libcublas.so.9.0:无法打开共享对象文件:没有这样的文件或目录

     

无法加载本机TensorFlow运行时。

     

请参阅   https://www.tensorflow.org/install/install_sources#common_installation_problems

     

出于某些常见原因和解决方案。包括整个堆栈跟踪   寻求帮助时,此错误消息上方。

我已经尝试使用ubuntu 16.04了,但是GPU不支持。 也不支持安装CUDA9.0和CUDA9.2。

我如何使用tensorflow-gpu?

我已经在〜/ .bashrc中应用了路径

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64$LD_LIBRARY_PATH

1 个答案:

答案 0 :(得分:1)

tensorflow-gpu软件包是针对Cuda 9.0构建的,但是您已安装Cuda 10.0。

您需要将Cuda的版本降级到9.0(但我记得,使用2080Ti是不可能的),或从源代码构建张量流。 tensorflow webiste

上有大量有关如何执行此操作的文档

您也可以尝试安装软件包tf-nightly-gpu。您应该注意,该版本是更具实验性的,因为尚未经过广泛的测试。