gpt4 book ai didi

gpu - Tensorflow 不使用 GPU

转载 作者:行者123 更新时间:2023-12-04 14:42:26 27 4
gpt4 key购买 nike

我正在尝试通过 Tensorflow 运行示例 seq2seq,但它不会使用 GPU。以下是我在带有 Tesla K20x 的 Linux 系统上安装 Tensorflow 所采取的步骤

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?任何帮助表示赞赏。

谢谢你。

最佳答案

问题似乎是当您 bazel run翻译示例,它在没有 GPU 支持的情况下重建。尝试添加 --config=cudabazel run命令,如下:

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

如果没有这个选项,Bazel 将在没有 GPU 支持的情况下重新编译整个 TensorFlow 运行时,并在运行示例应用程序时使用这个版本。

关于gpu - Tensorflow 不使用 GPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33970755/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com