python 大量文本数据预处理（NLP），跑的很慢，是否可以用上GPU加速

Question

我现在有一批数据要进行文本预处理，主要是NER（但是调某个服务的接口，所以也不是本地用模型跑的），但现在问题是数据量很大，跑的很慢，能不能用gpu让数据并行运算（但数据都是文本），因为我本人之前对gpu并行运算完全不懂，所以现在有点儿摸不着头脑，希望专业人士能给点儿建议，谢谢。

代码就是调用某服务接口，以及自己写的词典匹配和否定检测，都不涉及深度学习模型

Answer 1

可以用啊，判断GPU是否可用 ，否则使用cpu

device = torch.device("cuda:0"  if torch.cuda.is_available() else "cpu")