ほぼ身内向けですが忘れっぽい人のために。。。
Google Colaboratoryにアクセス
colab上でコマンドを打つときは先頭に!を付ける必要があります。
- 新規python3ノートブックを作る
- pythonのバージョン確認
!python --version
- cudaのバージョン確認
!ls /usr/lib/x86_64-linux-gnu/libcuda*
- 確認できたらPyTorchのダウンロードページにてダウンロード方法を選択
- PyTorchをインストール
一例:!pip3 install torch torchvision
- インポートできるか確認
import torch