PYTHON调用GPU之HelloWorld
PYTHON调用GPU之HelloWorld
import cv2
import numpy as np
from numba import cuda
import time
import math
import os
# os.environ['CUDA_VISIBLE_DEVICES'] = "0"
@cuda.jit
def process_gpu(img,channels):
tx = cuda.blockIdx.x*cuda.blockDim.x + cuda.threadIdx.x
原创
2021-05-16 22:21:22 ·
604 阅读 ·
0 评论