利用Google Colaboratory提供的GPU进行深度学习模型的训练

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上运行实验。

你可能感兴趣的:(利用Google Colaboratory提供的GPU进行深度学习模型的训练)