2016-10-24 23 views
0

im2txtが0.5GBのGeForce 970 GTXを使用するのを避ける方法があるかどうかを知りたかっただけです。通常の3.5GBだけを使用するトレーニングセッションを強制できますか?いいアイデアですか?Tensorflow - im2txtでのGeForce 970でのメモリ管理

ありがとうございました。

+0

不運であれば、それは遅い0.5ギガバイトを含めることができたものの割り当てメモリ私はあなたが使用する特定のGPUメモリを選択することができるとは思わないではありません。 970には、そのメモリパーティションに関するいくつかの問題があるようです。 GPUの使用を無効にすることができます。 – fabrizioM

+1

問題の深刻なアプリケーションのGPUを無効にすることは意味がありません。 –

+0

重大なアプリケーションにGeForce 970を使用するのは理にかなっていません。 (バギー)ゲームカードです。あなたのお金を取り戻す:http://www.pcgamer.com/heres-how-to-claim-your-30-nvidia-geforce-gtx-970-settlement/ – fabrizioM

答えて

1

以下のコマンドを使用してメモリの合計量を制限することができます。必ず

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) 
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) 
+0

ありがとう、私は家に帰るとすぐにそれを試してみます。 –