search

zhang0jhon / AttentionOCR

Scene text recognition
834 stars 259 forks source link

Memory usage increasing with increase in available GPU memory #73

Open shekargoud26 opened 4 years ago

shekargoud26 commented 4 years ago

I have run the provided docker image in 2 machines with 2 different GPUs

  1. GTX 970m 4GB
  2. Tesla V100 32GB

Its using around 220mb in 970m and 790mb on V100

here is the screenshot of 970m

gtx970m

here is the screenshot of V100

v100

both running on Ubuntu. Is there any way to fix the amount of GPU memory used by this api?

encoding-muzzu commented 4 years ago

yeah same here, even when i run in teslav100 , its occupying almost 8GB of memory, is there any way to reduce the memory?

image