search

NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library
Other
4.01k stars 450 forks source link

pynvml Issue #56

Closed msaad1311 closed 3 years ago

msaad1311 commented 3 years ago

Hello. I installed the imaginaire library as instructed in the installation guide and when running the fs-vid2vid i am getting the following error

pynvml.nvml.NVMLError_LibraryNotFound: NVML Shared Library Not Found

Can you kindly guide the way forward in this case? I am using Windows 10 with the same CUDA and cuDNN version as mentioned in the Installation guide.

arunmallya commented 3 years ago

You can try to remove all calls to this file: https://github.com/NVlabs/imaginaire/blob/master/imaginaire/utils/gpu_affinity.py This is the only one that uses pynvml