这是我的代码，我正在一个大数据集（37k行）上应用这个函数，我想在多线程上运行，或者用其他方法让它更快，我已经测试了Spark和Dask库，但是我陷入了无法解决的错误。如果你们有任何想法，那就太好了。

import matplotlib.pyplot as plt

def caption_from_image_file(x):
    return [str(get_caption(i,device)) for i in x.load()]

import cv2
import numpy as np

df = dg.getData("train")

df_test = df

# start timer  
import time
start_time = time.time()

df_test['captions'] = df_test.images.apply(caption_from_image_file)

# end timer (in minutes)
print("--- %s minutes ---" % ((time.time() - start_time)/60))

df_test.to_csv('test.csv',index=False)

# # free up cuda memory
torch.cuda.empty_cache()

df_test.captions