5
Tôi đang cố gắng chạy ví dụ seq2seq bởi Tensorflow, nhưng nó sẽ không sử dụng GPU. Dưới đây là bước tôi đã thực hiện để cài đặt Tensorflow trên hệ thống Linux với Tesla K20xTensorflow không sử dụng 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
Sau khi tất cả bước này, tôi đã cài đặt tensorflow. Sau đó, tôi cố gắng chạy ví dụ seq2seq, bởi
bazel run -c opt //tutorials/models/rnn/translate:translate
nhưng nó sẽ không sử dụng GPU. sau đó tôi thử ví dụ
bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu
và nó mang lại cho một lỗi
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
Đã tự hỏi nếu có ai biết những gì có thể gây ra các chương trình không sử dụng GPU? Bất kỳ trợ giúp được đánh giá cao.
Cảm ơn bạn.
bạn sẽ cần phải chạy ./configure bên trong thư mục tensorflow. Sau đó, bạn có thể đề cập đến con đường của người lái xe cuda của bạn và đừng quên đề cập đến phiên bản tương thích cuda. –