CUDA矢量化装饰器：[100]调用cuInit会导致CUDA_ERROR_NO_DEVICE：

Question

我正在测试我的GPU，并使用了示例cuda代码。不幸的是，它不断给我一个错误。

我尝试将CUDA_VISIBLE_DEVICES env变量设置为0，但没有用

import numpy as np
from numba import vectorize, cuda
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
@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)

我希望输出是2个numpy数组的加法而没有任何错误