在Google Colab上运行GPU

Google Colab research能提供在线编译的GPU环境,其实就是内嵌了一个Notebook。打开Google Colab,新建一个笔记本,在notebook setting里面选择GPU就可以了。

其中主要注意以下几点:

  1. pip安装的句子格式有少许变化:
!pip install -q keras
  1. 【打开文件】现在通常是打开Google云盘里的文件:地址前要加一个‘drive/’
with open('drive/review30000.csv') as f:
    reviewfile=pd.read_csv(f)

官方教程里的方法只有打开Google sheets和开发者关联云存储的方法,但没有提到怎么直接打开云盘文件,后来看了CSDN一位博主才知道。

【体验结果】运行相同实验,GPU比在本地直接训练每个epoch快了一倍左右,棒棒的~


image.png

其他使用colab进行实验的案例:

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容