geo-deep-learning
geo-deep-learning copied to clipboard
pynvml error

Pynvml dependency is used in torch an error arises in a new environment without pynvml installed
Originally posted by @valhassan in https://github.com/NRCan/geo-deep-learning/pull/344#discussion_r986328985
Same here with a new environment...
Torch version: pytorch 1.12.1 py3.10_cuda11.6_cudnn8.3.2_0 pytorch