CUDA_OUT_OF_MEMORY_ERROR

原因：其他python程序(很可能是tf的gpu程序)正在占用gpu

解决：关掉其他python gpu程序,只在terminal跑代码