最近时间发现,如果用本机跑图片分类的话,实在是跑不动,跑得太慢了,服务器也不快。之前在搞了一通cuda、cudnn之后,还是觉得慢,而且cuda对MacOS太不友好了。
在师兄推荐下尝试了一下google的colab。
进入了之后可以在设置里选择GPU或者TPU:
速度确实会快很多。但是我感觉colab有一个挺不方便的一点就是,本地文件需要上传,上传有两种方式。
第一个是在代码中输入:
from google.colab import files
uploaded = files.upload()
这样可以直接选择文件上传。
第二种方法是直接先将文件传到google drive。个人感觉这种方法更快。
其他的方面使用起来与jupyter notebook没差。
如果对速度和连接时间由更多的要求的话,可以购买colab pro版本,一个月9.9刀,价位也还挺合理的。速度是真的很香。
网友评论