RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB解决办法

    科技2024-07-21  66

    RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 768.50 MiB already allocated; 16.39 MiB free; 9.50 MiB cached)解决方法

    使用GPU出现上述错误,尝试改小batch_size的大小

    主要是记录一下自己平时遇到的问题,和大家分享一下 如有侵犯,请联系我 点个赞支持一下吧

    Processed: 0.008, SQL: 8