1. ホーム
  2. python

[解決済み] TensorflowをCPUで動作させる方法

2022-04-22 10:47:57

質問

Ubuntu 14.04にGPU版のtensorflowをインストールしました。

私は、tensorflow が利用可能な GPU にアクセスできる GPU サーバーにいます。

CPUでtensorflowを動かしたいのですが。

通常、私は env CUDA_VISIBLE_DEVICES=0 を使えば、GPU No. 0.

代わりにCPUを選ぶにはどうしたらいいのでしょうか?

でコードを書き換えることに興味はありません。 with tf.device("/cpu:0"):

どのように解決するのですか?

を適用することができます。 device_count パラメータは tf.Session :

config = tf.ConfigProto(
        device_count = {'GPU': 0}
    )
sess = tf.Session(config=config)

protobuf設定ファイルも参照してください。

tensorflow/core/framework/config.proto