メモ書き:pytorch

インストール

は~い、今日はPyTorch入れまぁす。
わざわざブログに残しているのは将来的に私が「pytorchってどうやって入れるんだっけ????」とか言い出すからです。


まずここにアクセスします。
pytorch.org

自分にあった環境を選んでください。
私はpip使っているのでpipで。

CUDAのバージョンを確認してください。
CMD使って「nvcc -V」で確認できましゅ。
10.1です。

選択肢にないんだが?????????????????????????????
pytorch.org
過去のバージョン見て適当に探してください。


見つけた(、もしくはRun this Commandに出てきた)コマンドを実行します。



おわり

その他

GPU使ってるか確認するやつ

torch.cuda.is_available()

FalseならGPU使ってない