Colaboratory是Google Drive中的一个工具。最近发现,它可以提供GPU用于深度学习模型的训练。这对于没有本地GPU的同学是一个好消息。
这篇文献介绍了如何使用Colaboratory创建一个Python脚本(Jupyter notebook),并创建深度学习环境,以及对模型进行训练的教程。其中提供的样例脚本可以从这里找到。
该脚本运行显示,成功设置了一块Tesla K80
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 4489883971363085120, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 356515840
locality {
bus_id: 1
}
incarnation: 14163943019613271790
physical_device_desc: "device: 0, name: Tesla K80, pci bus id: 0000:00:04.0, compute capability: 3.7"]
后续任务:将自己的训练脚本放到Colaboratory上运行实验。