首页 文章

在ubuntu18.04中导入错误tensorflow-gpu

提问于
浏览
0

我在导入tensorflow时遇到一些错误 .

我的电脑规格:

操作系统:ubuntu 18.04

Nvidia RTX 2080 Ti * 2

Nvidia司机-415

CUDA:10.0

cuDNN:7.3.0 tensorflow:1.11.0

import tensorflow

错误:

Traceback(最近一次调用最后一次):文件“/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py”,第58行,来自tensorflow.python.pywrap_tensorflow_internal import * File“/ 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,pathname,description)文件”/usr/lib/python3.6/imp.py“,第243行,在load_module中return load_dynamic(name,filename,file)文件“/usr/lib/python3.6/imp.py”,第343行,在load_dynamic中返回_load(spec)ImportError:libcublas.so.9.0:无法打开共享对象文件:否这样的文件或目录在处理上述异常期间,发生了另一个异常:Traceback(最近一次调用last):文件“”,第1行,在文件“/usr/local/lib/python3.6/dist-packages/tens orflow / init.py“,第22行,来自tensorflow.python import pywrap_tensorflow #pylint:disable = unused-import文件”/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行,in raise ImportError(msg)ImportError:Traceback(最近一次调用last):文件“/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py”,第58行,来自tensorflow.python.pywrap_tensorflow_internal import * File“/ 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,pathname,description)文件”/usr/lib/python3.6/imp.py“,第243行,在load_module中返回l oad_dynamic(name,filename,file)文件“/usr/lib/python3.6/imp.py”,第343行,在load_dynamic中返回_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中使用了app路径

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 回答

  • 1

    tensorflow-gpu 包是针对Cuda 9.0构建的,但您安装了Cuda 10.0 .

    您需要将您的Cuda版本降级到9.0(但如果我记得,使用2080Ti是不可能的),或者从源代码构建张量流 . 有关如何在tensorflow webiste上执行此操作的大量文档

    您也可以尝试安装包 tf-nightly-gpu . 你应该注意到这个版本更具实验性,因为它没有经过广泛的测试 .

相关问题