RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0； 10.76 GiB total capacity； 7.64

错误原因：

使用了CUDA_VISIBLE_DEVICES指定了空闲的3090显卡，但仍报错，原因是batchsize设置得太高，设置了128.

解决方案：

降低到64即可。

或者使用两张及以上的卡。

版权声明：本文为qq_45128278原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。