zhangliliang / RPN_BF

RPN+BF
Other
203 stars 75 forks source link

Is GTX980 suitable for this test #22

Closed Yezipiaomu closed 7 years ago

Yezipiaomu commented 7 years ago

My computer has the nvidia graphic card which is GTX980 but when I run the demo you given, always out of memory,can you give me some advice to solve the problem ? Thank you

zhangliliang commented 7 years ago

Hi,

The RAM of 980 is 4 GB, and it might be not enough for running the VGG model. Might be help to use the CuDNN for saving the memory usage.

Yezipiaomu commented 7 years ago

I don't know if I have made the mistake,when I run the demo,It always is out of memory!!! could you tell me the mininus Graphic RAM that I need!!!

zhangliliang commented 7 years ago

Actually, it needs a GPU whose RAM is 8G or more (without the help of the CuDNN), since the VGG model is RAM consuming.