如何借助GPU提升效率？

阿智兄

非常感谢你在百忙之中抽空为我解答。
以下代码作用是在EXCEL表ABCDE列中选三列随机抽取4行做比较，由于效率很低，如果想通过调用GPU的并行运算去提高效率，该如何改呢？

1. import xlrd
   
2. import multiprocessing as mp
   
3. from functools import partial
   
4. import csv
   
5. import random as rd
   
6. 
   
7. def read_file(file):
   
8.     data=xlrd.open_workbook(file)
   
9.     names=data.sheet_names()
   
10.     res=[]
    
11.     for i in names:
    
12.         table=data.sheet_by_name(i)
    
13.         rows=table.nrows
    
14.         res.append([table.row_values(j) for j in range(rows)])
    
15.     res=[[[j[0],j[1],j[2],tuple(j[3].split('A')[1:3]),j[4],j[3]] for j in i] for i in res]
    
16.     return(res,names)
    
17. 
    
18. def write_file(file,res,names,maxlength):
    
19.     head=[k for j in [[i]+['']*4 for i in names] for k in j]
    
20.     f=open(file,'w',newline='')
    
21.     csvw=csv.writer(f)
    
22.     csvw.writerow(head)
    
23.     for i in range(maxlength):
    
24.         csvw.writerow([k for j in [data[i][:3]+data[i][:-3:-1] for data in res] for k in j])
    
25.     f.close()
    
26. 
    
27. def check(sample,data2,index,around):
    
28.     res=[]
    
29.     for i in sample:
    
30.         if around!=-1:
    
31.             temp=[j  for j in data2 if i[index[0]]==j[index[0]] and i[index[1]]==j[index[1]] and abs(i[index[2]]-j[index[2]])<=around]
    
32.         else:
    
33.             temp=[j  for j in data2 if i[index[0]]==j[index[0]] and i[index[1]]==j[index[1]] and i[index[2]]==j[index[2]]]
    
34.         if len(temp):
    
35.             res.append(temp[0])
    
36.         else:
    
37.             break
    
38.     else:
    
39.         return(res)
    
40.     return([])
    
41.                
    
42. def cal(label,data,index,num,around):
    
43.     if len(data[label])>=num:
    
44.         sample=rd.sample(data[label],num)
    
45.     else:
    
46.         sample=data[label]
    
47.     temp=[check(sample,i,index,around) for i in data[:label]+data[label+1:]]
    
48.     if [j for i in temp for j in i]:
    
49.         return([sample]+temp)
    
50.     else:
    
51.         return([])
    
52. 
    
53. if __name__=='__main__':
    
54.     alphatdict=dict([[chr(i),i-65] for i in range(65,91)])
    
55.     file = "test.xlsx"
    
56.     index=sorted([alphatdict[i] for i in input('需要比对的三列：').upper()])
    
57.     if 4 in index:
    
58.         around=0
    
59.     else:
    
60.         around=-1
    
61.     num=4
    
62.     filenum=10000
    
63.     label=0
    
64.     data,names=read_file(file)
    
65.     ff=partial(cal,data=data,index=index,num=num,around=around)
    
66.     while label<filenum:
    
67.         pool=mp.Pool(mp.cpu_count())
    
68.         res=pool.map(ff,range(len(data)))
    
69.         pool.close()
    
70.         pool.join()
    
71.         res=[[k for j in i for k in j] for i in res]
    
72.         maxlength=max([len(i) for i in res])
    
73.         res=[i+[['']*6]*(maxlength-len(i)) for i in res]
    
74.         if sum([len(i) for i in res]):
    
75.             write_file('res'+str(label+1)+'.csv',res,names,maxlength)
    
76.             label+=1
    
77.         else:
    
78.             pass
    

复制代码

赚小钱

关键词 cuda，具体可以搜索: "cuda 编程" , “cuda cpp 编程", "cuda python 编程”

阿智兄

有哪位高人可以指点一下吗？