2017-11-12 8 views
-1

mscocoデータセットcodeを使ってfaster_rcnn_with resnet101でgooglesオブジェクト検出を訓練したいと思います。私は訓練目的のために10,000のイメージしか使用しなかった。私はグラフィックを使用した:GeForce 930M/PCIe/SSE2。 NVIDIAドライバのバージョン:384.90。ここに私のGeForceの写真があります。 enter image description hereシステムRAMからテンソルフローのGPUメモリを伸ばす方法

私は8GbのRAMを持っていますが、テンソルフローのgpuでは1.96Gbです。

system configurationterminal show of GPU。 PGUのRAMをどのように拡張できますか?私は完全なシステムメモリを使いたい。

答えて

0

CPUを訓練して、マシンのRAMを利用することができます。しかし、gpu上で何かを実行するには、最初にgpuにロードする必要があります。今では、メモリをスワップアウトすることができます。なぜなら、すべての結果がどのステップでも必要なわけではないからです。しかし、非常に長いトレーニング時間をかけてお支払いいただき、バッチサイズを減らすようお勧めします。それにもかかわらず、このプロセスと実装に関する詳細は、https://medium.com/@Synced/how-to-train-a-very-large-and-deep-model-on-one-gpu-7b7edfe2d072にあります。

+0

私は既にCPUで動いています。しかし、私はGPUで走りたいと思っています。なぜテンソルフローGPUは1.96GBですか?私はあなたが私のシステムのRAM(8GB)を見たと確信しています@トーマスPinetz – orbit

+0

gpuは、独自のメモリを使用し、あなたはわずか2 GBのメモリを持つGPUを持っています。あなたがgpu上のものを計算したい場合は、gpuにデータをロードし、gpuメモリにロードする必要があります。おそらく、cudaカーネルと、それをよりよく理解するためにgpuにデータをロードする方法を見てください。 –

関連する問題