Question

我使用以下命令安装了Tensorflow：

$ pip install tensorflow-gpu

安装看起来很好，但是当我尝试在python中导入tensorflow时，我收到以下错误消息：

ImportError: Traceback (most recent call last):
  File "/home/ep15/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/ep15/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/ep15/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: /home/ep15/.local/lib/python2.7/site-packages/tensorflow/python/../libtensorflow_framework.so: undefined symbol: cuDevicePrimaryCtxSetFlags

我的操作系统是Ubuntu 14.04。我使用的是CUDA 8.0.1

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2013 NVIDIA Corporation
Built on Wed_Jul_17_18:36:13_PDT_2013
Cuda compilation tools, release 5.5, V5.5.0

相反，如果我按照此处的说明操作：https://www.tensorflow.org/versions/r0.12/get_started/os_setup

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.1-cp27-none-linux_x86_64.whl

安装有效，但它是旧版本的tf（0.12）。

编辑：看起来我只能安装到1.0.0版本，但我无法安装任何更高版本。

Tensorflow安装问题版本1.4.1

0 个答案: