2015-12-10 13 views
16

Tensorflowユニットテストを手動で実行する方法はありますか? TFのソースコードを変更しながら健全性チェックを実行したいTensorflowユニットテストを実行する

多くのテスト操作を実行するクラスを持つ多くの_test.pyファイルがあり、実行方法を理解できません。簡単な方法があるはずですか?

答えて

22

TensorFlowのユニットテストを実行する最も簡単な方法は、Bazelを使用して、あなたがdownloaded the source from Gitを持っていると仮定している。

# All tests (for C++ changes). 
$ bazel test //tensorflow/... 

# All Python tests (for Python front-end changes). 
$ bazel test //tensorflow/python/... 

# All tests (with GPU support). 
$ bazel test -c opt --config=cuda //tensorflow/... 
$ bazel test -c opt --config=cuda //tensorflow/python/... 
+0

そして、どのように、あなたは、GPUをサポートしてこれらのテストを実行しますか?それらを直接実行するだけで、CPUを使用するように見えます: 'I tensorflow/core/common_runtime/local_device.cc:40]ローカルデバイス内並列処理スレッド:16 テンソルフロー/コア/ common_runtime/direct_session.cc:58]ダイレクトセッションインターオペラ並列スレッド:16 – panmari

+0

GPUサポートの構築方法と実行方法を示す答えを更新しました。 (これらのログメッセージは、CPU *と* GPUモードで生成されることに注意してください。) – mrry

+0

確かに、私はただCPUが現在リストされていることを強調したいと思います。あなたの更新をありがとう! – panmari

関連する問題