由于我的数据集很大,因此我尝试在colab上使用GPU。我正在使用NLTK进行某些文本处理。即使我选择“运行时”->“更改运行时类型”->“硬件加速器”->“ GPU”,我的代码运行速度仍然很慢,并且我收到一条警告,指出“您没有使用GPU,您想切换到标准加速器吗?” 我的代码如下:
def preprocess(element):
### convert all to lowercase ###
element = element.lower()
### get rid of punctuation ###
element = element.translate({ord(i): None for i in string.punctuation})
### get rid of stopwords ###
element = [i for i in element if i not in stopwords.words('english')]
return element
final['text'] = final.text.map(preprocess)
其中final是需要预处理的带有“文本”列的数据框。我尝试过:
!pip install thinc_gpu_ops
import thinc_gpu_ops
thinc_gpu_ops.AVAILABLE
它给我输出True 如果我在这里缺少东西,请告诉我。