[Feature]: How to Release GPU Memory Occupation

towhee-io / towhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.

https://towhee.io

Apache License 2.0

3.16k stars 246 forks source link

[Feature]: How to Release GPU Memory Occupation #2690

Closed Git-xu closed 6 months ago

Git-xu commented 7 months ago

Is there an existing issue for this?

[X] I have searched the existing issues.

Is your feature request related to a problem? Please describe.

发现每次使用都会导致gpu内存占用一直在升高且不会自动释放掉

Describe the solution you'd like.

No response

Describe an alternate solution.

No response

Anything else? (Additional Context)

No response

junjiejiangjjj commented 7 months ago

Use torch.cuda.empty_cache() to release GPU memory

stale[bot] commented 6 months ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions. Rotten issues close after 30d of inactivity. Close the stale issues and pull requests after 7 days of inactivity. Reopen the issue with /reopen.