Tensorflow nicht mit GPU

Ich versuche, führen Sie das Beispiel seq2seq von Tensorflow, aber es wird nicht die GPU nutzen. Hier sind die Schritt, den ich nahm, um zu installieren Tensorflow auf einem Linux-system mit Tesla K20x

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

Nachdem alle von diesem Schritt habe ich tensorflow installiert. Ich versuche dann zu laufen, der seq2seq Beispiel durch

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

aber es wird nicht die GPU. Dann versuche ich das Beispiel 

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

und es gibt einen Fehler, 

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

Frage mich, ob jemand weiß, was möglicherweise dazu, dass das Programm nicht verwenden, um nicht die GPU? Jede Hilfe ist willkommen.

Danke.

  • Sie würde ausführen müssen ./konfigurieren von innen tensorflow Ordner. Dann können Sie vergessen, Ihren cuda-Treiber ist Weg und vergessen Sie nicht zu erwähnen die cuda-Kompatibilität version.
InformationsquelleAutor user108372 | 2015-11-28
  1. 5

    Das Problem sieht, dass, wenn Sie bazel run die übersetzung Beispiel, er erstellt ohne GPU-Unterstützung. Versuchen Sie --config=cuda zu den bazel run Befehl wie folgt:

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

    Ohne diese option Bazel wird, kompilieren Sie die gesamte TensorFlow Laufzeit ohne GPU-Unterstützung, und verwenden Sie diese version, wenn es läuft die Beispiel-Anwendung.

    InformationsquelleAutor mrry
  2. 1

    es tritt auf, da der cuda ist nicht richtig verbunden. Geben Sie den folgenden Befehl in das terminal -

    sudo ldconfig /usr/local/cuda/lib64

    InformationsquelleAutor ram
  3. 1

    Ich vermute, sollten Sie es mit einer GPU-version 

    pip install tensorflow-gpu
    InformationsquelleAutor BenjiBB
Schreibe einen Kommentar