Skip to content

因為顯存不夠,只有16GB,想詢問是否能使用v1.5的版本? #335

@HaoHaohedgehog

Description

@HaoHaohedgehog

如果能使用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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions