-
Notifications
You must be signed in to change notification settings - Fork 795
Open
Description
如果能使用v1.5版本的話,應該如何調整?在網站上未尋找到解決或更改方法。謝謝。
以下為報錯:
gradio.exceptions.Error: 'Error during processing: CUDA out of memory. Tried to allocate 2.00 GiB. GPU 0 has a total capacity of 15.70 GiB of which 1.42 GiB is free. Process 406442 has 14.28 GiB memory in use. Of the allocated memory 8.34 GiB is allocated by PyTorch, and 5.57 GiB 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 (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)'
Metadata
Metadata
Assignees
Labels
No labels