Google Colab 如何设置GPU运行?

作者: 托尼 分类: AI,Cloud Server,Python 发布时间: 2019-04-02 15:37

第一次用Google Colab ,主要是奔着免费的Tesla K80 而去。但是你写了一段代码却会发现根本就没有GPU。这是为什么? 因为默认情况下是不启用GPU的,如果想要使用必须手动设置。而且每个.ipynb创建后都得重新设置。

 

设置方法如下

1 点击上方的【代码执行程序】

2. 点击【更改运行是类型】

 

3.  下拉选中【GPU】

在笔记本设置页面,可以选择Python版本,分别是Python2和Python3。 硬件加速器,默认情况下是None, 问好前面有个小三角,点一下就可以选择GPU。

如果你的代码支持TPU,就可以选择TPU选项。 这个没有试过,但是未来也许都会用这个。

如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!

打赏      

发表评论

邮箱地址不会被公开。 必填项已用*标注