mberr / torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA
https://torch-max-mem.readthedocs.io/en/latest/
MIT License
14 stars 0 forks source link

Updates for torch 2.0 #9

Closed mberr closed 1 year ago

mberr commented 1 year ago

This PR updates the package to torch 2.0. For sake of code simplicity, it drops support of torch < 2.0. It also cleans up some CI config.