为什么colab pro的GPU跑的很慢-解决方法

最近在colab上跑了一下cifar-10的图像分类数据,结果发现跑的很慢。拿本机的CPU试了一下,一个epoch大概需要20min;在colab的GPU上甚至需要两倍以上的时间。感觉很不合常理。

先确认了一下是否在使用GPU:


OK,确实是有在用cuda。百思不得其解之下,在Stack Overflow上发现了一个帖子:
https://stackoverflow.com/questions/60798910/google-colab-pro-gpu-running-extremely-slow

跟我的问题很像,帖子中有个回复说drive是个云盘(我的数据挂在到了google drive上),每次读取数据都比较慢,可以拷贝到local路径或者直接下载到local中。


我直接下载到了当前路径,解压:

!mkdir train_local
!wget https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
!tar zxvf cifar-10-python.tar.gz -C train_local
!rm cifar-10-python.tar.gz

重新训练,速度快了非常多(原来一个epoch大概需要6min,现在只需要20sec。

但是这里有个疑惑是,都用云盘读取的时候,用colab的GPU仍然比CPU(类型“None”)要慢?

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章