geo-deep-learning
geo-deep-learning copied to clipboard
Replace Pynvml utilities with torch.cuda's utils
Torch.cuda... utilities has everything we need to replace the pynvml utilities and dependency.
2 examples:
- device count: pynvml vs torch.cuda
- get free memory: pynvml vs torch.cuda