Python-CUDA отключен из-за установки NUMBA_DISABLE_CUDA = 1 в среде

Question

Это первый раз, когда я пытаюсь использовать cuda в python, поэтому я ознакомился с некоторыми уроками, касающимися того, как его настроить ... но я закончил с этой ошибкой:

numba.cuda.cudadrv.error.CudaSupportError: Error at driver init: 
CUDA is disabled due to setting NUMBA_DISABLE_CUDA=1 in the environment, or because CUDA is unsupported on 32-bit systems.:

Я пользуюсь версией x64 для Windows, и я также проверил версию на python (которая также есть на x64).Я использую Nvidia GTX 960m с последними драйверами.

Я пытался найти эту проблему более двух недель, но не нашел решения, которое могло бы работать

Также я вставлю код здесь, но я не думаю, что здесь есть проблема:

@vectorize(["float32(float32, float32)"], target='cuda')
def test(a,b):
    return a+b

def main():

    a=np.arange(1,10,dtype=np.float32)
    b=np.arange(11,20,dtype=np.float32)
    c=np.zeros(9, dtype=np.float32)

    c=test(a,b)

    print(c)

main()

Answer 1

Я запускаю [sic] версию для x64 Windows, и я [sic] также проверил версию на python (которая также есть на x64).Я использую Nvidia GTX 960m с последними драйверами.

Несмотря на это утверждение, ваш интерпретатор Python является 32-разрядным.Это было подтверждено в комментариях к print(tuple.__itemsize__), возвращающему 4. Для 64-битного интерпретатора Python вернулось бы 8.

Здесь вы можете ясно видеть *1009*, что NUMBA_DISABLE_CUDA можно установить только в Numbaесли он установлен в среде хоста или с использованием 32-битного интерпретатора Python, с определяется значением tuple.__itemsize__ в Numba.Это последний случай.

Я пытался найти эту проблему уже более двух недель, но не нашел решения, которое могло бы работать

Решения не существует, потому что любая версия CUDA, которая является достаточно новой для работы с Numba, не поддерживает 32-битные среды в Windows.Это давно устарело на платформах Windows.Единственный выход - установить собственную 64-битную версию Anaconda с 64-битным интерпретатором Python и соответствующими библиотеками.