繁体   English   中英

使用Cuda 10安装pip tensorflow-gpu后出现错误

[英]Error after installing pip tensorflow-gpu with cuda 10

如果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]键入“帮助”,“版权”,“信用”或“许可证”以获取更多信息。

>>>导入张量流

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

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

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

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

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

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

点安装tensorflow-gpu == 1.13.0rc0

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

问题是您尚未完全满足TensorFlow GPU要求。
完整名单是这里的TF网站上。

错误的主要原因似乎是您没有使用cuDNN SDK正确配置CUDA工具包9.0 ,请注意您需要的版本大于或等于7.2,并将其添加到系统路径中。

我正在运行python 3.7并能够通过运行cuda 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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM