Я получаю это сообщение:
"Невозможно найти библиотеку драйвера CUDA. Если вы уверены, что драйвер CUDA установлен, попробуйте установить переменную среды NUMBAPRO_CUDA_DRIVER с путем к файлу общего драйвера CUDAбиблиотека. "
Я не уверен, как изменить переменную среды, как это предлагается.Это делается с помощью некоторого типа команды, подобной этой?:
$ export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
Я почти уверен, что CUDA установлена правильно, отправка nvcc -V возвращает это:
"nvcc: NVIDIA (R) Драйвер компилятора Cuda Copyright (c) 2005-2015 NVIDIA Corporation Построен на Tue_Aug_11_14: 27: 32_CDT_2015 Инструменты компиляции Cuda, выпуск 7.5, V7.5.17 "
Я переустановил и попытался следовать этим инструкциям https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#mandatory-post как можно лучше несколько раз, но теперь у меня нет идей ...
Вот код, который я пытаюсь выполнить:
import numpy as np
from numba import vectorize
@vectorize(["float32(float32, float32)"], target='cuda')
def VectorAdd(a,b):
return a + b
N = 1000
A = np.ones(N, dtype=np.float32)
B = np.ones(N, dtype=np.float32)
C = np.ones(N, dtype=np.float32)
C = VectorAdd(A,B)