前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >为什么colab pro的GPU跑的很慢-解决方法

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

作者头像
生信编程日常
发布2020-11-26 12:11:51
6K0
发布2020-11-26 12:11:51
举报
文章被收录于专栏:生物信息学、python、R、linux

最近在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中。

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

代码语言:javascript
复制
!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”)要慢?

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档