python - 找不到 libNVVM

Question

我正在尝试通过 CUDA @vectorize 一些 python。我正在使用 Cuda9.1、Anaconda3 在 atom 中运行 Python3，并且遇到与此类似的问题:

Cuda: library nvvm not found

但该解决方案适用于 linux 而我在 windows 上。

我从上面的解决方案中了解到我需要添加一个环境变量。我试图通过搜索我在以下位置找到的 nvvm 文件来做到这一点:

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

但添加它并没有帮助。我尝试了“conda install cudatoolkit”并进行了更新。

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

Answer 1

我在 2019 年底在没有 Anaconda 的 CUDA 10.1 上遇到了这个问题(所有东西都是通过 pip 安装的)。麻麻documentation说它首先查找由名为 cudatoolkit 的 conda 包配置的 CUDA，但这不适用，因为我没有使用 conda 安装任何东西。它寻找 CUDA 的第二个地方是在环境变量 CUDA_HOME 中，并将其设置为“C:\

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