OOM when allocating tensor with shape

在调试程序的时候碰到这个问题，显卡内存不够，分析原因主要有两个：

1，batch_size太大，内存不够。

2，显卡内存被别的程序占用没有退出，可以在后台查看占用线程并终止。