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