python代码在gpu上面运行_如何将python代码转换为在GPU上而不是CPU上运行?

我得到了一个python代码,它接收两个图像作为输入,并使用Gabor过滤器来查找两个图像的RGB的相关性,并将其保存在csv文件中。所以我需要使用GPU来执行程序,因为它需要很多时间和CPU利用率。我有一个geforcegtx1050ti和一个完全的编程初学者。在

我做了一些研究,了解了CUDA和Tensorflow,但是我真的不确定如何继续实现它,以及在不改变代码的情况下实现它的最佳方法是什么。在#Gabor Filter

def build_filters():

filters = []

#tesing phrase filter - reduce

for ksize in range(9, 19, 5):

for theta in np.arange(45, 225, 45):

for sigma in range(2,6,2):

kern = cv2.getGaborKernel((ksize, ksize), sigma, theta, 5.0, 0.5, 0, ktype=cv2.CV_32F)

kern /= 1.5*kern.sum()

filters.append(kern)

return filters

#Apply filter into the image

def process(images, f):

accum = np.zeros_like(images)

for kern in f:

fimg = cv2.filter2D(images, cv2.CV_8UC3, kern)

np.maximum(accum, fimg, accum)

return accum

谢谢你!在

你可能感兴趣的:(python代码在gpu上面运行_如何将python代码转换为在GPU上而不是CPU上运行?)