Как мне сделать этот процесс на графическом процессоре, чтобы улучшить скорость поиска? - PullRequest
Новая
       3

Как мне сделать этот процесс на графическом процессоре, чтобы улучшить скорость поиска?

1 голос
/ 15 января 2020

У меня есть вектор размером 512, и я хочу сравнить его с 30 000 аналогичных элементов.

Как мне выполнить этот процесс на графическом процессоре для повышения скорости поиска? Это часть кода, который выполняется на графическом процессоре и замедляет выполнение всего кода, поскольку поиск передается на процессор.

Мой python код для загрузки набора данных: 

all_files = glob.glob("/home/hbr-ubuntu/qqww/Dataset/*.out")
f3=[]
n3=[]
for t in all_files:
    f2 = np.loadtxt(t,dtype='float32',delimiter=',' )
    f3.append(f2)
    t = t.replace("/home/hbr-ubuntu/qqww/Dataset/", "")
    t = t.replace(".out", "")
    n3.append(t)

Поиск кода: 

for r in f3:         
        sim = cp.dot(r, img_feature.T)
        if sim > 0.4:
        print('sim is', sim)

С наилучшими пожеланиями.

...