Imageai: Custrom detection GPU RAM required for different batch sizes

Created on 6 Aug 2019  路  2Comments  路  Source: OlafenwaMoses/ImageAI

Currently on my local GPU with 4gb of RAM I am unable to use batch sizes greater than 2. That appears to use about 3GB of RAM. Does the RAM use increase with batch size linearly? Or otherwise does anyone have an estimate of the amount needed for batch sizes of 4 or 8? The tutorial mentions the k80 with 24GB of RAM as I understand.

question
Source
picture nineclicks

Most helpful comment

On my RTX2070 8GB I can confirm I cannot go higher than batch_size=6.

picture Meulen92 on 19 Aug 2019
👍2

All 2 comments

On my RTX2070 8GB I can confirm I cannot go higher than batch_size=6.

Meulen92 picture Meulen92 on 19 Aug 2019
👍2

Thank you, this answers my question.

nineclicks picture nineclicks on 19 Aug 2019
👍1
Was this page helpful?
0 / 5 - 0 ratings

Related issues

huangl6 picture huangl6  路  6Comments
Combinacijus picture Combinacijus  路  3Comments
bestbugwriter picture bestbugwriter  路  5Comments
NikolayTV picture NikolayTV  路  5Comments
aless2003 picture aless2003  路  5Comments