使用Cuda 10安装pip tensorflow-gpu后发生错误

时间:2019-02-01 09:25:52

标签: python tensorflow gpu

如果tensorflow-gpu的错误代码在cpu上运行,我想只使用conda版本中的tensorflow的pip版本。

在我导入tensorflow时为ubuntu 18.0.4安装cuda 10和cudnn后,出现以下错误。

PS:我清除了与cuda和Nvidia有关的所有内容,然后安装了新的Cuda 10。

  

Python 3.6.7 | Anaconda自定义(64位)| (默认值,2018年10月23日,19:16:44)   Linux上的[GCC 7.3.0]   输入“帮助”,“版权”,“信用”或“许可证”以获取更多信息。

     >>>导入张量流      

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

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

  

回溯(最近通话最近):     文件“”,第1行,位于     在第24行的文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/init.py”中       从tensorflow.python import pywrap_tensorflow#pylint:disable = unused-import     在第49行的文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/init.py”中       从tensorflow.python导入pywrap_tensorflow     在第74行中输入“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py”       提高ImportError(msg)   ImportError:追溯(最近一次呼叫最近):     在第58行的“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py”文件中       从tensorflow.python.pywrap_tensorflow_internal导入*     在第28行的文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py”中       _pywrap_tensorflow_internal = swig_import_helper()     swig_import_helper中的第24行,文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py”       _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路径名,描述)     在load_module中的文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/imp.py”,第243行       返回load_dynamic(名称,文件名,文件)     在load_dynamic中的文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/imp.py”,第343行       返回_load(spec)   ImportError:libcublas.so.9.0:无法打开共享库文件:没有这样的文件或目录   
  ImportError:libcublas.so.9.0:无法打开共享对象文件:没有这样的文件或目录

我不知道为什么会出现此错误,我没有cuda verison 9

3 个答案:

答案 0 :(得分:4)

可通过pip获得的tensorflow-gpu的当前稳定版本是v1.12,它是根据CUDA 9.0构建的。如果您查看tensorflow github存储库的问题编号22706,则预期即将发布的tensorflow-gpu v1.13的官方二进制文件将针对CUDA 10构建。

自2019年1月23日起,已针对CUDA 10构建了rc0 version of tensorflow-gpu v1.13。可以通过指定以下版本使用pip进行安装:

  

pip install tensorflow-gpu == 1.13.0rc0

由于系统中的CUDA 9已被CUDA 10取代,默认的tensorflow-gpu代码失败,因为它的版本为1.12,并期望系统中存在CUDA 9二进制文件。因此,当前,如果您想在CUDA 10中使用tensorflow,那么最快的解决方案就是按照上面指定的方式升级tensorflow版本。

答案 1 :(得分:1)

问题是您尚未完全满足TensorFlow GPU要求。
完整列表为TF网站上的here

错误的主要原因是您的CUDA toolkit 9.0cuDNN SDK的配置不正确,请注意,您需要的版本大于或等于7.2,并将其添加到系统路径中。

答案 2 :(得分:0)

我正在运行python 3.7,并能够通过运行:cud 10安装tensorflow:

 pip install --no-cache-dir https://github.com/evdcush/TensorFlow-wheels/releases/download/tf-1.12.0-py37-gpu-10.0/tensorflow-1.12.0-cp37-cp37m-linux_x86_64.whl