要注意是 Out Of Memory 还是 CUDA_ERROR_OUT_OF_MEMORY
    Out Of Memory 是内存不足
    CUDA_ERROR_OUT_OF_MEMORY 是显存不足
    参见 各GPU机型内存显存大小 选择更大内存或更大显存的机型使用，前提是确保您的程序本身不存在内存泄漏的问题