使用Google Colaboratory免费GPU

简介

Google Colab,全名Colaboratory,可以用它来提高Python技能,也可以用Keras、TensorFlow、PyTorch、OpenCV等等流行的深度学习库来练练手,开发深度学习应用。

转载:薅资本主义羊毛,用Google免费GPU

准备工作

在Google Drive上创建ipynb笔记本

Colab用的数据都存储在Google Drive云端硬盘上,所以,我们需要先指定要在Google Drive上用的笔记本。

设置免费GPU

选择代码执行程序中的更改运行时类型

硬件加速器选择GPU

然后,Google Colab就可以用了。

用Colab运行.py文件

先运行下面这些代码,来安装必要的库、执行授权。

1
2
3
4
5
6
7
8
9
10
11
12
!apt-get install -y -qq software-properties-common python-software-properties module-init-tools
!add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
!apt-get update -qq 2>&1 > /dev/null
!apt-get -y install -qq google-drive-ocamlfuse fuse
from google.colab import auth
auth.authenticate_user()
from oauth2client.client import GoogleCredentials
creds = GoogleCredentials.get_application_default()
import getpass
!google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
vcode = getpass.getpass()
!echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}

运行的时候应该会看到下图所示的结果:

看见那个链接之后,点击它,复制验证码并粘贴到文本框里。

授权完成后,就可以挂载Google Drive了:

1
2
!mkdir -p drive
!google-drive-ocamlfuse drive

之后,就可以把本地的程序和数据上传到Google Drive上,再用Colab访问运行。

试着跑一下《Tensorflow实战》里AlexNet的一段代码:

快得一笔,和我本地跑的CPU比较一下:

缺点应该是每次重开.ipynb都要重新挂载下Google Drive,另外,据说连续使用时间不能超过12小时。

一分一毛,也是心意。