无法找到libNVVM

Question

我试图通过CUDA @vectorize一些python.我用Cuda9.1,Anaconda3在原子中运行Python3,并且遇到类似的问题:

Cuda:未找到图书馆nvvm

但是那个解决方案适用于linux而且我在Windows上.

我从上面的解决方案中收集,我需要添加一个环境变量.我试着通过从我找到的nvvm文件中搜索来做到这一点:

C:\用户\ ME \应用程序数据\本地\程序\ Python的\ Python36\LIB \站点包\numba\CUDA \测试\ cudadrv__pycache__

但补充说没有帮助.我试过'conda install cudatoolkit'以及更新它.

我的问题是操作系统不知道这个库在哪里？如何将python指向它需要的库？

Answer 1

这些是我设置的环境变量,用于在Windows上运行一些pyCUDA示例代码:

NUMBAPRO_NVVM = C:\ Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1 \nvvm\bin \nvvm64_32_0.dll

NUMBAPRO_LIBDEVICE = C:\ Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1 \nvvm\libdevice \

Answer 2

我在 2019 年底在没有 Anaconda 的 CUDA 10.1 上遇到了这个问题（所有东西都是通过 pip 安装的）。numba文档说它首先查找由名为cudatoolkit的 conda 包配置的 CUDA ，但这并不适用，因为我没有使用 conda 安装任何东西。它查找 CUDA 的第二个位置是环境变量 CUDA_HOME，将其设置为“C:\Program Files\NVIDIA GPU Compute Toolkit\CUDA\v10.1”对我来说是成功的：

此后，错误消失并且 numba 代码运行。