百度Aistudio 中使用gpu的方法

import numpy as np
 
from numba import vectorize
 
@vectorize(['float32(float32, float32)'], target='cuda')
 
def Add(a, b):
 
    return a + b
 
# Initialize arrays
 
N