TensorFlowではなく、PyTorchはいかがでしょうか？という気分。
環境の確認
CoLabのNotebook上で、python --versionしてみたら、3.6.3だった。
CUDAのバージョンとかがよく分からんけれど、PyTorchのサイトに従って、pip install -U torch torchvisionする。
意外と、あっさりインストールできた。
チュートリアル実行
そのまま、チュートリアルを、実行してみる。
図のように、trainingも普通に動く。
本日の結論
GoogleColaboratory上でも、PyTorchをGPU版で動かすことができました、というハナシ。