python - 如何让 CUDA 与 python 3.5 和 numba 一起工作

Question

通过运行代码：

import numba

numba.cuda.api.detect()

我得到错误：

Traceback (most recent call last):
  File "<pyshell#1>", line 1, in <module>
    numba.cuda.api.detect()
  File "C:\Users\Wakasuki\AppData\Local\Programs\Python\Python35-32\lib\site-packages\numba\cuda\api.py", line 257, in detect
    print('Found %d CUDA devices' % len(devlist))
  File "C:\Users\Wakasuki\AppData\Local\Programs\Python\Python35-32\lib\site-packages\numba\cuda\cudadrv\devices.py", line 49, in __len__
    return len(self.lst)
  File "C:\Users\Wakasuki\AppData\Local\Programs\Python\Python35-32\lib\site-packages\numba\cuda\cudadrv\devices.py", line 26, in __getattr__
    numdev = driver.get_device_count()
  File "C:\Users\Wakasuki\AppData\Local\Programs\Python\Python35-32\lib\site-packages\numba\cuda\cudadrv\driver.py", line 269, in get_device_count
    self.cuDeviceGetCount(byref(count))
  File "C:\Users\Wakasuki\AppData\Local\Programs\Python\Python35-32\lib\site-packages\numba\cuda\cudadrv\driver.py", line 218, in __getattr__
    self.initialization_error)
numba.cuda.cudadrv.error.CudaSupportError: Error at driver init: 
CUDA disabled by user:

如何在 numba 中启用 CUDA？我刚刚在 python 3.5 中安装了 CUDA 工具包 7.5 和最新的 numba 和 numpy。

Answer 1

尝试如下：

'...' 是“nvvm64_31_0.dll”和“libdevice”的 CUDA 工具包 7.5 的安装路径

import os
os.environ['NUMBAPRO_NVVM']=r'C:\NVIDIA GPU Computing Toolkit\CUDA\v8.0\nvvm\bin\nvvm64_31_0.dll'
os.environ['NUMBAPRO_LIBDEVICE']=r'C:\NVIDIA GPU Computing Toolkit\CUDA\v8.0\nvvm\libdevice'