使用cuda报错的一次记录(CUDA error: out of memory)

 原因:

由于batch_size设置过大导致的!!!

猜你喜欢

转载自blog.csdn.net/wtl1992/article/details/131668678