Google ColaboratoryでのGPUの使用について

Question

Google colabでのGPU使用について
「ランタイムのタイプを変更」からGPUを使用できる設定にすれば、計算は自動的にGPUが行うようになるのでしょうか？

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

のようにGPUに渡すコードを書く必要はないのでしょうか？
よろしくお願い致します。

Answer 1

GPUに渡すコードを書く必要はないのでしょうか？

TensorFlow なら GPU 版の TensorFlow がインストールされていて、GPU が利用可能な状況であれば、自動で GPU が利用されます。
Pytorch なら GPU を利用する場合、質問のコードのように使用するデバイスオブジェクトを作成し、Tensor.to() でテンソルの計算を実行するデバイスに転送する必要があります。
つまり、GPU で実行したい場合はそのことを明示的に書く必要があります。

Comments

[n_thk72]

ありがとうございました！