Как получить CUDA для работы с питона 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: 

Как включить CUDA в Numba? Недавно я установил CUDA toolkit 7.5 и самые последние numba и numpy в python 3.5.

Answer 1

Try следующим образом:

'...' ваша установка путь CUDA Toolkit 7.5 "nvvm64_31_0.dll" и "libdevice"

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'