libNVVM cannot be found

Question

I am trying to @vectorize some python through CUDA. I'm running Python3 in atom with Cuda9.1, Anaconda3, and have a similar problem to this:

Cuda: library nvvm not found

but that solution is for linux and I'm on windows.

I gather from that solution above I need to add an Environment Variable. I tried to do just that by searching from the nvvm file which i found in:

C:\Users\ME\AppData\Local\Programs\Python\Python36\Lib\site-packages\numba\cuda\tests\cudadrv__pycache__

but adding that didn't help. I tried 'conda install cudatoolkit' as well as updating it.

Is my problem that the OS doesn't know where this library is? How to I point python to the library that it needs?

Answer 1

I had this problem in late 2019 on CUDA 10.1 without Anaconda (everything installed through pip). The numba documentation said that it looks for CUDA first as configured by a conda package called cudatoolkit, but that wasn't applicable as I hadn't installed anything using conda. The second place it looks for CUDA is in the environment variable CUDA_HOME, and setting this to "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1" did the trick for me:

Thereafter, the errors go away and numba code runs.