[Feature]: How to Release GPU Memory Occupation
Git-xu opened this issue · 2 comments
Git-xu commented
Is there an existing issue for this?
- 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
Use torch.cuda.empty_cache() to release GPU memory
stale commented
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.