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

【问题标题】：How to get CUDA to work with python 3.5 and numba如何让 CUDA 与 python 3.5 和 numba 一起工作
【发布时间】：2016-05-02 04:24:01
【问题描述】：

通过运行代码：

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。

【问题讨论】：

标签： python numba numba-pro

【解决方案1】：

尝试如下：

'...' 是“nvvm64_31_0.dll”和“libdevice”的CUDA toolkit 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'

【讨论】：