Adjustment needed in file 'gpus.py'

[VeroniqueClauwaert]

Please forgive me if I am not stating this in the correct way (first comment), but I had a recent issue with the koila package and found the solution as well. Namely, the import of the package was halted due to an incorrect import from pynvml.
The solution is changing line 6 in file 'gpus.py': 'from pynvml.smi import nvidia_smi' to 'from pynvml_utils.smi import nvidia_smi'.
Hope it can help somebody!

[rentruewang]

Huh I believe the package names have changed since I wrote it. Thanks for pointing it out. Would you like to file a PR and tag me as reviewer?

I do not advise using koila in production as it was just a proof of concept of what could be done to resolve out of memory errors automatically, and I do not actively maintain it or use it.