Python报错:00M: When allocating tensor with shape.

OOM : When allocating tensor with shape.

报错原因:显卡内存不够;

解决办法:①、首先查看显卡占用情况,有可能是显卡内存被别的程序占用没有退出,可以后台查看;

                  ②、排除第一种情况之后,考虑:bath_size可能设置过大,可以适当调小。

猜你喜欢

转载自blog.csdn.net/songchunxiao1991/article/details/82424219