Hi kayccc,
I have created a new topic there: cuDNN take up too much GPU memory Could you help to take a look? Thanks