how to improve the gpu training?

[McAlexDudu]

Hi, I tried GPU version of deepmd, since I just had one GPU device and only 2GMB memory. But when training, the memory seems use not enough, and I wonder if I could improve the speed?

[njzjz]

What do you mean when you said the memory not enough? As shown in your snapshot, only 488MB of memory was used.

[McAlexDudu]

Sorry, I didn't say it cleary. I mean how to use more GPU memory to impove the speed?

[njzjz]

The memory is related to your data size, model size, batch size, etc., but not related to performance.

[McAlexDudu]

Got it！