How to run python script on GPU

run_on_gpu.py

# pip install --user numba

from numba import jit, cuda 
import numpy as np 
# to measure exec time 
from timeit import default_timer as timer    
  
# normal function to run on cpu 
def func(a):                                 
    for i in range(10000000): 
        a[i]+= 1      
  
# function optimized to run on gpu  
@jit(target ="cuda")                          
def func2(a): 
    for i in range(10000000): 
        a[i]+= 1
if __name__=="__main__": 
    n = 10000000                            
    a = np.ones(n, dtype = np.float64) 
    b = np.ones(n, dtype = np.float32) 
      
    start = timer() 
    func(a) 
    print("without GPU:", timer()-start)     
      
    start = timer() 
    func2(a) 
    print("with GPU:", timer()-start) 

This doesn't work anymore ):

The difference between both functions seems to be only

@jit(target ="cuda")

This doesn't work anymore ):

Hey @FatinShadab try replacing @jit(target='cuda') with @jit(target_backend='cuda')

can i use it like:

# loop_mode = 'cuda' or 'cpu'
def looper(loop_mode):

    @jit(target_backend=loop_mode)
     for i in range (......

i mean can i use it like :
@jit(target_backend='cpu')