2016-08-04 189 views
0

我试图安装Tensorflow与CUDA支持。这里是我的规格:Tensorflow CUDA GTX 1070导入错误

  • NVIDIA GTX 1070
  • CUDA 7.5
  • Cudnn V5.0

我已经通过了pip安装安装Tensorflow - 所以我想象你的答案被从源代码安装,但我想确保没有快速修复。

的错误是:

[email protected]:~$ python 
Python 2.7.10 (default, Oct 14 2015, 16:09:02) 
[GCC 5.2.1 20151010] on linux2 
Type "help", "copyright", "credits" or "license" for more information. 
>>> import tensorflow as tf 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.so locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcudnn.so locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.so locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.so.1 locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.so locally 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/__init__.py", line 23, in <module> 
    from tensorflow.python import * 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/__init__.py", line 98, in <module> 
    from tensorflow.python.platform import test 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/test.py", line 77, in <module> 
    import mock    # pylint: disable=g-import-not-at-top,unused-import 
    File "/usr/local/lib/python2.7/dist-packages/mock/__init__.py", line 2, in <module> 
    import mock.mock as _mock 
    File "/usr/local/lib/python2.7/dist-packages/mock/mock.py", line 71, in <module> 
    _v = VersionInfo('mock').semantic_version() 
    File "/usr/local/lib/python2.7/dist-packages/pbr/version.py", line 460, in semantic_version 
    self._semantic = self._get_version_from_pkg_resources() 
    File "/usr/local/lib/python2.7/dist-packages/pbr/version.py", line 447, in _get_version_from_pkg_resources 
    result_string = packaging.get_version(self.package) 
    File "/usr/local/lib/python2.7/dist-packages/pbr/packaging.py", line 725, in get_version 
    raise Exception("Versioning for this project requires either an sdist" 
Exception: Versioning for this project requires either an sdist tarball, or access to an upstream git repository. Are you sure that git is installed? 

我正在从主目录中的python控制台 - 不是在Tensorflow目录。

GIT和CUDA都安装:

[email protected]:~$ git --version 
git version 2.5.0 
[email protected]:~$ nvcc -V 
nvcc: NVIDIA (R) Cuda compiler driver 
Copyright (c) 2005-2015 NVIDIA Corporation 
Built on Tue_Aug_11_14:27:32_CDT_2015 
Cuda compilation tools, release 7.5, V7.5.17 

我验证CUDA是通过本次测试功能(发现here):

/usr/local/cuda/bin/cuda-install-samples-7.5.sh ~/cuda-samples 
cd ~/cuda-samples/NVIDIA*Samples 
make -j $(($(nproc) + 1)) 

Tensorflow成功安装:

export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0rc0-cp27-none-linux_x86_64.whl 
sudo -H pip install --upgrade $TF_BINARY_URL 

我的GPU似乎很好:

[email protected]:~$ nvidia-smi 
Thu Aug 4 17:31:47 2016  
+-----------------------------------------------------------------------------+ 
| NVIDIA-SMI 367.35     Driver Version: 367.35     | 
|-------------------------------+----------------------+----------------------+ 
| GPU Name  Persistence-M| Bus-Id  Disp.A | Volatile Uncorr. ECC | 
| Fan Temp Perf Pwr:Usage/Cap|   Memory-Usage | GPU-Util Compute M. | 
|===============================+======================+======================| 
| 0 GeForce GTX 1070 Off | 0000:03:00.0  On |     N/A | 
| 0% 41C P8 12W/185W | 499MiB/8104MiB |  0%  Default | 
+-------------------------------+----------------------+----------------------+ 

+-----------------------------------------------------------------------------+ 
| Processes:              GPU Memory | 
| GPU  PID Type Process name        Usage  | 
|=============================================================================| 
| 0  900 G /usr/bin/X          272MiB | 
| 0  1679 G compiz           154MiB | 
| 0  2287 G ...s-passed-by-fd --v8-snapshot-passed-by-fd 69MiB | 
+-----------------------------------------------------------------------------+ 
+0

也许试试'pip install --upgrade setuptools' –

+0

'require already up to date' –

+0

听起来像某种版本不匹配,也许尝试在新env中安装virtualenv? –

回答

4

从错误日志看来,看起来像某种版本与mock或pbr软件包不匹配,可能来自以前的安装。在这种情况下,从源代码构建将无济于事,您需要的是从头开始安装Python依赖项,即通过在新环境中使用virtualenv安装。

8

其a bug in pbr。该缺陷描述中包含了解决出口PBR版本:

export PBR_VERSION=X.Y.Z

策略路由版本可以得到尽可能pbr -v

+0

这也解决了Jenkins Job Builder中的错误 – 79E09796

+0

我现在也发现了这个问题。有人能指出解决方案吗? – trialcritic

+0

就是这样。只要导出pbr版本,它应该是足够的 –