2015-11-23 3 views
2

Я только что прошел через cifar10 tensorflow tutorial на своем mac. Этот урок вызвал мой энтузиазм по поводу NN, поэтому я установил сервер Ubuntu для обучения в больших масштабах.python - выполнение тензорного потока на сервере Ubuntu с CUDA GeForce9600GT

Однако я не могу запустить cifar10_train.py на своем сервере. Я всегда запустить в эту ошибку:

Путь исходного tensorflow кода:

~/python/tensorflow/tensorflow/tensorflow/ 

Путь установки tensorflow virtualenv:

~/tensorflow/ 

Cmd:

source ~/tensorflow/bin/activate #activate virtualenv 
python/tensorflow/tensorflow/tensorflow/models/image/cifar10/cifar10_train.py #the raw source code of tensorflow is in ~/python/tensorflow/tensorflow/tensorflow 

Ошибка:

Traceback (most recent call last): 
    File "python/tensorflow/tensorflow/tensorflow/models/image/cifar10/cifar10_train.py", line 28, in <module> 
    import tensorflow.python.platform 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 4, in <module> 
    from tensorflow.python import * 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 22, in <module> 
    from tensorflow.python.client.client_lib import * 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/client/client_lib.py", line 35, in <module> 
    from tensorflow.python.client.session import InteractiveSession 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 11, in <module> 
    from tensorflow.python import pywrap_tensorflow as tf_session 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module> 
    _pywrap_tensorflow = swig_import_helper() 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 24, in swig_import_helper 
    _mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description) 

Я установил tensorflow так же, как на мой макинтош в virtualenv и активировал его правильно перед выполнением сценария. Как и многие другие темы, я предложил обновить уже шесть, однако у меня по-прежнему такая же ошибка.

Update 1 После прохождения этого вопроса темы на GitHub/tensorflow через Я заметил, что это ошибка, связанная с CUDA. Я добавил это в моем пути среды:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64" 
export CUDA_HOME=/usr/local/cuda 

Я по-прежнему работать в ошибки, однако он сократил до следующего:

Traceback (most recent call last): 
File "cifar10_train.py", line 28, in <module> 
import tensorflow.python.platform 
ImportError: No module named tensorflow.python.platform 

Update 2 Как кто-то предложил, установить Protobuf через пип. Ошибка в очередной раз изменить по какой-то странной причине:

Traceback (most recent call last): 
    File "cifar10_train.py", line 28, in <module> 
    import tensorflow.python.platform 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 4, in <module> 
    from tensorflow.python import * 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 13, in <module> 
    from tensorflow.core.framework.graph_pb2 import * 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/core/framework/graph_pb2.py", line 16, in <module> 
    from tensorflow.core.framework import attr_value_pb2 as tensorflow_dot_core_dot_framework_dot_attr__value__pb2 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/core/framework/attr_value_pb2.py", line 16, in <module> 
    from tensorflow.core.framework import tensor_pb2 as tensorflow_dot_core_dot_framework_dot_tensor__pb2 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/core/framework/tensor_pb2.py", line 16, in <module> 
    from tensorflow.core.framework import tensor_shape_pb2 as tensorflow_dot_core_dot_framework_dot_tensor__shape__pb2 
    File "/home/it13095/tensorflow/local/lib/python2.7/site-packages/tensorflow/core/framework/tensor_shape_pb2.py", line 22, in <module> 
    serialized_pb=_b('\n,tensorflow/core/framework/tensor_shape.proto\x12\ntensorflow\"d\n\x10TensorShapeProto\x12-\n\x03\x64im\x18\x02 \x03(\x0b\x32 .tensorflow.TensorShapeProto.Dim\x1a!\n\x03\x44im\x12\x0c\n\x04size\x18\x01 \x01(\x03\x12\x0c\n\x04name\x18\x02 \x01(\tb\x06proto3') 
TypeError: __init__() got an unexpected keyword argument 'syntax' 

ответ

5

TensorFlow в настоящее время требует CUDA Toolkit 7.0 и cuDNN.

Для cuDNN требуется cc 3.0 GPU, а для CUDA toolkit 7.0 требуется cc 2.0 GPU.

Ваш 9600GT не удовлетворяет эти requirements:

In order to build or run TensorFlow with GPU support, both Cuda Toolkit 7.0 and CUDNN 6.5 V2 from NVIDIA need to be installed.

TensorFlow GPU support requires having a GPU card with NVidia Compute Capability >= 3.5.

Так что, если вы хотите использовать TensorFlow с поддержкой GPU, вы будете нуждаться в cc3.5 или выше GPU, и следуйте инструкциям, чтобы установить необходимую поддержку программного обеспечения. Кроме того, вы можете установить TensorFlow без поддержки графического процессора.

Смежные вопросы