Resource exhausted: OOM when allocating tensor with shape[50,50] and type float on /job:localhost/re

preds = model.predict_proba(X, batch_size=512)

应该是gpu内存不够了。如果可以的话，可以调低batch_size。若仍然不行，那么可以通过减小处理的数据量来解决。因为显存的大小，决定了它的一个运行过程能够处理的数据量的大小。而设置不同的batch_size，可以加快或者降低处理的速度。当然了，把batch_size设置为一个更小的值，则可以在一定程度上弥补显存的不足，但这只能治标不治本。