Python代码无法访问GPU CUDA驱动程序

Question

当我尝试执行试图调用GPU的Python测试脚本（计算功能6和正确安装的CUDA /功能正确的CUDA 8）时，CLI会标记错误。据我所知，Python（32位）已正确安装（对于64位Windows OS），并且可以正常工作。 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而被禁用，   或因为32位系统不支持CUDA。