最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

google colaboratory - How to fix memory issue with A100 in colab - Stack Overflow

programmeradmin4浏览0评论

Every time i run my code in colab with A100/L4 GPU/T4 GPU i am getting below error. Can someone help

OutOfMemoryError: CUDA out of memory. Tried to allocate 500.00 MiB. GPU 0 has a total capacity of 22.16 GiB of which 201.38 MiB is free. Process 26183 has 21.96 GiB memory in use. Of the allocated memory 21.77 GiB is allocated by PyTorch, and 9.19 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (.html#environment-variables)

发布评论

评论列表(0)

  1. 暂无评论