安装pip tensorflow-gpu且cuda10后错误

2024-10-01 13:40:55 发布

您现在位置:Python中文网/ 问答频道 /正文

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

在为ubuntu18.0.4安装cuda10和cudnn之后,当我导入tensorflow时,它会给出以下错误。在

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

Python 3.6.7 |Anaconda custom (64-bit)| (default, Oct 23 2018, 19:16:44) [GCC 7.3.0] on linux Type "help", "copyright", "credits" or "license" for more information.

>>> import tensorflow

Traceback (most recent call last): File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in from tensorflow.python.pywrap_tensorflow_internal import * File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in _pywrap_tensorflow_internal = swig_import_helper() File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) File "/home/purav/anaconda3/envs/purav/lib/python3.6/imp.py", line 243, in load_module return load_dynamic(name, filename, file) File "/home/purav/anaconda3/envs/purav/lib/python3.6/imp.py", line 343, in load_dynamic return _load(spec) ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

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

Traceback (most recent call last): File "", line 1, in File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/init.py", line 24, in from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/init.py", line 49, in from tensorflow.python import pywrap_tensorflow File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 74, in raise ImportError(msg) ImportError: Traceback (most recent call last): File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in from tensorflow.python.pywrap_tensorflow_internal import * File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in _pywrap_tensorflow_internal = swig_import_helper() File "/home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) File "/home/purav/anaconda3/envs/purav/lib/python3.6/imp.py", line 243, in load_module return load_dynamic(name, filename, file) File "/home/purav/anaconda3/envs/purav/lib/python3.6/imp.py", line 343, in load_dynamic return _load(spec) ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

我不知道为什么会有这个错误我没有cuda verison 9


Tags: inpyimporthomelibtensorflowlinesite
3条回答

目前通过pip提供的tensorflow-gpu的稳定版本是v1.12,它是针对cuda9.0构建的。{cutensda的正式版本是针对即将发布的gpv1.10或tensda存储库的。在

截至2019年1月23日,rc0 version of tensorflow-gpu v1.13已经发布,它是针对cuda10构建的。可以使用pip安装,方法是指定以下版本:

pip install tensorflow-gpu==1.13.0rc0

由于系统中的CUDA 9已替换为CUDA 10,因此默认的tensorflow gpu代码将失败,因为它是版本1.12,并且希望CUDA 9二进制文件出现在系统中。所以目前,如果你想在CUDA10中使用tensorflow,那么最快的解决方案就是升级上面指定的tensorflow版本。在

问题是您没有完全满足TensorFlow GPU的要求。
完整的名单在TF网站上here

错误的主要原因似乎是您没有使用cuDNN SDK配置好CUDA toolkit 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

相关问题 更多 >