Stable Diffusion OutOfMemoryError: CUDA out of memory. Tried to allocate 2.00 GiB
时间: 2023-10-23 22:29:58 浏览: 185
遭遇OutOfMemoryError
这个错误是由于CUDA内存不足引起的。错误信息中指出尝试分配2.00 GiB的内存,但GPU的总容量为7.93 GiB,已经分配了758.34 MiB,剩余5.75 MiB,总共保留了858.00 MiB的内存。
为了解决这个问题,有几个可能的解决方案。首先,你可以尝试设置`max_split_size_mb`参数来避免内存碎片化。另外,你需要确认本地文件和远程文件是否同步,并检查代码中是否指定了正确的GPU操作。
此外,你还可以检查你选择的GPU卡号是否正确。你可以使用`import os`和`os.environ["CUDA_VISIBLE_DEVICES"] = "0"`来确保选择了正确的GPU卡号。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [报错记录torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 90.00 MiB (GPU 0; 7.93 ...](https://blog.csdn.net/gongkeguo/article/details/127674524)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文