2017-01-23 2 views
1

Я недавно установил NVIDIA CUDA и пробный запуск кода:Как решить ошибку CUDA PATH?

from numba import cuda 
from numba import * 
import numpy as np 
from pylab import imshow, show 
from timeit import default_timer as timer 

Но я получил эту ошибку в убунту 16.0.4. Как я могу решить эту ошибку:

@cuda.jit(argtypes=[f8, f8, f8, f8, uint8[:,:], uint32]) 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/decorators.py", line 92, in kernel_jit 
    kernel.bind() 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/compiler.py", line 489, in bind 
    self._func.get() 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/compiler.py", line 366, in get 
    cuctx = get_context() 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/devices.py", line 194, in get_context 
    return _runtime.get_or_create_context(devnum) 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/devices.py", line 162, in get_or_create_context 
    return self.push_context(self.gpus[devnum]) 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/devices.py", line 40, in __getitem__ 
    return self.lst[devnum] 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/devices.py", line 26, in __getattr__ 
    numdev = driver.get_device_count() 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/driver.py", line 307, in get_device_count 
    self.cuDeviceGetCount(byref(count)) 
    File "/usr/local/lib/python2.7/dist-packages/numba/cuda/cudadrv/driver.py", line 248, in __getattr__ 
    self.initialization_error) 
CudaSupportError: Error at driver init: 

CUDA driver library cannot be found. 
If you are sure that a CUDA driver is installed, 
try setting environment variable NUMBA_CUDA_DRIVER 
with the file path of the CUDA driver shared library. 

ответ

1

Попробуйте ввести следующий переменные

export NUMBAPRO_NVVM=/home/cuda-7.5/nvvm/lib64/libnvvm.so 
export NUMBAPRO_LIBDEVICE=/home/cuda-7.5/nvvm/libdevice/ 

в файле .bashrc, который обычно находится в домашнем/имени пользователя/ В вашем случае путь выше, должен быть исправлен в соответствии с вашими путями установки cuda. Обратите внимание, что даже если вы используете просто umba, а не numbapro, имена переменных должны быть такими, как показано выше.

Смежные вопросы