web-dev-qa-db-ja.com

SSE pipを使用した指示を使用したTensorflowのインストール

提供されているデフォルトの指示 here を使用して、ubuntu 16.04にcpu only tensorflowを正常にインストールしました。この説明では、virtualenvとpipを使用することを推奨しているため、ソースからビルドしませんでした。私はこれらの指示でインストールに問題がありませんでした。

同じページにある さらに下 の説明を使用してインストールを検証しました。プログラムが正常に実行されている間、次の警告が出力されました。

W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE3 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
Hello, TensorFlow!

私が調べたすべてのスタックオーバーフローページは、bazelを使用してソースからビルドした人にのみ言及していますが、pipを使用した人には当てはまらないようです。

SSEの指示に従って、インストールを再コンパイルまたは更新するにはどうすればよいですか?

11
CuriousDan

bazel--config=optオプションを使用して、アーキテクチャに合わせてカスタマイズされたホイールを構築し、pipを使用して、生成されたホイールをインストールする必要があります。

2014 MacBook + Xeon V3に最適化されたビルドを https://github.com/yaroslavvb/tensorflow-community-wheels に時々アップロードする予定です

Bazelをインストールしたら、pipコマンドのホイールを作成するのはおおよそ次のようになります

./configure

export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
export flags="--config=opt --config=cuda -k"

bazel build $flags -k //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

次に、ホイールを/tmp/tensorflow_pkgにドロップします

3
bazel build --linkopt='-lrt' -c opt --copt=-mavx --copt=-msse4.2 --copt=-msse4.1 --copt=-msse3-k //tensorflow/tools/pip_package:build_pip_package

上記で問題が解決した場合は、投票してくださいChristian Frei's回答 これらのテンソルフロー警告を解決するにはどうすればよいですか

1
Ajay Singh

これらは単なる警告です。この link に従って、TensorFlowをインポートする前にこの行を追加し、

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
import tensorflow as tf

私も同じ問題を抱えていますが、提案されたリンクで問題が解決しました。

この answer も見てください。SSE情報と、さまざまなタイプのログインジケーターについて詳しく知るのに役立ちます。

0
Krunal Kapadiya