无法在colab上使用GPU

时间:2020-04-30 05:02:49

标签: google-colaboratory

由于我的数据集很大,因此我尝试在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 如果我在这里缺少东西,请告诉我。

0 个答案:

没有答案