[BUG] Increasing GPU memory
Open
@vmoens is already working on this.
Since Jan 14, 2023.
bug
- Dominant language
- Python
- Stars
- 3.6k
- Forks
- 487
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 207
Description
Describe the bug
The GPU memory keeps increasing if we do not explicitly collect the generation using gc.collect
To Reproduce
Example is here
Reason and Possible fixes
This happens due to mujoco rendering using GPU. Can we handle this internally in torchRL?
Checklist
- [ x] I have checked that there is no similar issue in the repo (required)
- [ x] I have read the documentation (required)
- I have provided a minimal working example to reproduce the bug (required)
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.