Когда я пытаюсь выполнить тестовые сценарии Python, которые пытаются вызвать графический процессор (возможность вычисления 6 и правильно установленный / функциональный CUDA 8), CLI отмечает ошибку. Насколько я могу судить, Python (32-разрядный) установлен правильно (для 64-разрядной ОС Windows) и в остальном работает нормально. Numba (0.44.1), pycuda (2019.1) и т. Д. Были установлены через pip (для Python 3.7.2), и все зависимости, по-видимому, установлены (самые последние версии). Никаких изменений не было внесено ни в какие установленные файлы. Я не пробовал добавлять что-либо в PATH или CLI, чтобы учесть это. Возможно, поможет установка последних версий драйверов (как рекомендовано в документации Numba).
Код работает, если 'cuda' заменен на 'cpu'.
import numpy as np
from numba import vectorize
@vectorize(['float32(float32, float32)'], target='cuda')
def Add(a, b):
return a + b
# Initialize arrays
N = 100000
A = np.ones(N, dtype=np.float32)
B = np.ones(A.shape, dtype=A.dtype)
C = np.empty_like(A, dtype=A.dtype)
# Add arrays on GPU
C = Add(A, B)
print(C)
Ожидаемый результат (при просмотре через CPU): [2. 2. 2. ... 2. 2. 2.]
Сообщение об ошибке (усечено):
numba.cuda.cudadrv.error.CudaSupportError: Ошибка при инициализации драйвера: CUDA
отключен из-за установки NUMBA_DISABLE_CUDA = 1 в среде,
или потому что CUDA не поддерживается в 32-битных системах .: