RuntimeError: CUDA out of memory. Tried to allocate 76.00 MiB (GPU 0; 23.70 GiB total capacity; 21.9

这是GPU显存不够了

解决方法1:换服务器

解决方法2:把batch size调小

也有可能是要用的卡被人占了

你可能感兴趣的:(python)