Sorry for confusion! 
It was like this :
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
Data has fixed shape, and I used RNN model to train.
I didn’t checked the memory usage of GPU or tried to run code with random input, so let me check it and come back!
Thank you very, very much!