torch-max-mem
torch-max-mem copied to clipboard
Decorators for maximizing memory utilization with PyTorch & CUDA
https://github.com/mberr/torch-max-mem/actions/runs/15103471625/job/42448022137 ``` FileNotFoundError: [Errno 2] No such file or directory: 'uvx' ```
the improvements to pykeen in https://github.com/pykeen/pykeen/pull/975 don't work now that we have totally externalized AMO to this package - can we think about how to make sure that the CPU...
This is an autogenerated PR. Use this to reject the changes in this repository. [Cruft](https://cruft.github.io/cruft/) has detected updates from the Cookiecutter repository.
This is an autogenerated PR. Use this to merge the changes to this repository. [Cruft](https://cruft.github.io/cruft/) has detected updates from the Cookiecutter repository.