CPU/GPU間のテンソルフロー

Question

テンソルフローGPUを搭載している（NVIDIA GeForce 950上で動作している）ので、CPUと性能を比較したいと思います。

私はtensorFlow MNISTチュートリアルコードを実行していますが、とにかく推定されたスピードが劇的に上昇していることに気づきました（2日前に、バッチサイズ100のラップトップi7でCPUバージョンを実行しました。 GPU、バッチサイズ10） - 私が切り替えたときのCPUとGPUの間...しかし、GPUのバッチサイズを100から10に下げたときの速度向上に気付いただけです...

今私は私が得ているものに対する客観的な尺度が欠けています。

CPUとGPUテンソルの間を切り替える方法はありますか？通常

unset CUDA_VISIBLE_DEVICES 

Answer 1

は、CPUにtf.deviceを設定してみてください：0

with tf.Session() as sess: 
    with tf.device("/cpu:0"): 

Answer 2

に戻るには

export CUDA_VISIBLE_DEVICES="" 

GPUを非表示にするには

Answer 3

別のオプションは、CPUのバージョンをインストールすることです2つの仮想環境でのテンソルフローのgpuバージョン、仮想環境にテンソルフローをインストールする方法の詳細な手順ここにリストされていますhttps://www.tensorflow.org/get_started/os_setup;このようにして、同じコードをCPUを使用する2つの端末ウィンドウで実行し、もう1つはGPUを使用することができます。