2016-09-08 4 views
9

読書を使用します https://www.tensorflow.org/versions/r0.10/resources/faq.htmlそれは述べている:設定Tensorflowは、すべてのCPUの

は、私のマシン上の 利用可能なすべてのデバイス(のGPUとCPU)を利用するTensorFlowていますか?

TensorFlowは、複数のGPUとCPUをサポートします。 TensorFlowが操作をデバイスに割り当てる方法の詳細については、TensorFlowでのGPUの使用に関するハウツー のドキュメント、および複数のGPUを使用するサンプルモデルのCIFAR-10チュートリアル を参照してください。

TensorFlowは計算能力が で3.5を超えるGPUデバイスのみを使用します。

Tensorflowは、指定されたマシン上のすべてのCPUを自動的に使用することができますか、明示的に構成する必要はありませんか?

+0

はそれがすべての利用可能なCPU – fabrizioM

+2

@fabrizioMのおかげを使用しますが、これはどこのドキュメントで参照されている:あなたはより多くを必要とするように感じる場合は、スレッドの数を制御することができますか? –

答えて

16

CPUは単にスレッドプールである「デバイス」を介して使用されます。

sess = tf.Session(config=tf.ConfigProto(
    intra_op_parallelism_threads=NUM_THREADS)) 
+0

このアプリケーションへのPythonのGILの制限について心配する必要はありませんか? GILのためにCPUをスレッディングするのは役に立たないのですか? – nikpod

+3

内部C++ランタイムが使用するスレッドの数で、Pythonとは独立しています。 –

+1

面白いです。何らかの理由でこれを1に設定すると、私の練習スピードが大幅に向上しました! – Zuoanqh

関連する問題