如何使用 CUDA 10.0 安装 tensorflow-gpu 1.12

Question

我知道 tensorflow-gpu==1.12 需要 CUDA 9.0 或更低版本，但是我可以在 CUDA 10.0 上安装 Tensorflow 吗？可能是通过来源或使用 Bazel?

更新：我尝试从 Github 下载并通过 Bazel 安装，但出现以下错误。

Python 3.7.1 (default, Oct 23 2018, 19:19:42) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
Traceback (most recent call last):
  File "/some-folder/tensorflow-1.12.0/tensorflow/python/platform/self_check.py", line 25, in <module>
    from tensorflow.python.platform import build_info
ImportError: cannot import name 'build_info' from 'tensorflow.python.platform' (unknown location)

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/some-folder/tensorflow-1.12.0/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "/some-folder/tensorflow-1.12.0/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/some-folder/tensorflow-1.12.0/tensorflow/python/pywrap_tensorflow.py", line 25, in <module>
    from tensorflow.python.platform import self_check
  File "/some-folder/tensorflow-1.12.0/tensorflow/python/platform/self_check.py", line 27, in <module>
    raise ImportError("Could not import tensorflow. Do not import tensorflow "
ImportError: Could not import tensorflow. Do not import tensorflow from its source directory; change directory to outside the TensorFlow source tree, and relaunch your Python interpreter from there.

Answer 1

tensorflow 不支持 python 3.7 版本。需要 Python 3.4、3.5 或 3.6

见下文 https://www.tensorflow.org/install/pip

Answer 2

我们需要从源目录和运行 python 文件中取出。如果您运行使用 python filename.py 命令创建文件，您需要将文件移出源目录。

当您从源目录使用 import tensorflow 时，它将导入文件夹 tensorflow 而不是实际的 tensorflow 库

如何使用 CUDA 10.0 安装 tensorflow-gpu 1.12

How to install tensorflow-gpu 1.12 with CUDA 10.0

open-source

deep-learning

tensorflow