Tensorflow не использует GPU

Я пытаюсь запустить пример seq2seq от Tensorflow, но он не будет использовать GPU. Вот шаг я взял установить Tensorflow на системе Linux с Tesla K20xTensorflow не использует GPU

git clone --recurse-submodules https://github.com/tensorflow/tensorflow 
./configure # Yes GPU 
bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer 
bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu # The GPU is being used) 
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg 
pip install /tmp/tensorflow_pkg/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

После всего этого шага, я tensorflow установлен. Затем я попытаюсь запустить пример seq2seq:

bazel run -c opt //tutorials/models/rnn/translate:translate

но он не будет использовать GPU. Затем я попробовать пример

bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu

и выдает ошибку

bazel-bin/tensorflow/cc/tutorials_example_trainer: error while loading shared libraries: /path/to/home/.cache/bazel/_bazel_hduong/9e8a6e75473e7bf5c9d1c8a084e2a0e9/tensorflow/bazel-out/local_linux-opt/bin/tensorflow/cc/../../_solib_local/_U_S_Sthird_Uparty_Sgpus_Scuda_Ccudart___Uthird_Uparty_Sgpus_Scuda_Slib64/libcudart.so.7.0: file too short

было интересно, если кто-нибудь знает, что может привести к программе, чтобы не использовать GPU? Любая помощь приветствуется.

спасибо.

источник

2015-11-28 user108372

вам нужно будет запустить ./configure внутри папки tensorflow. Затем вы можете указать путь вашего драйвера cuda и не забудьте указать версию совместимости cuda. –

Проблема заключается в том, что когда вы приводите пример перевода, он восстанавливается без поддержки графического процессора. Попробуйте добавить --config=cuda к команде bazel run, следующим образом:

$ bazel run -c opt --config=cuda //tensorflow/models/rnn/translate:translate

Без этой опции Базэл будет перекомпилировать весь TensorFlow выполнения без поддержки GPU, и использовать эту версию, когда он запускает пример приложения.

источник

2015-12-17 18:23:11 mrry