PyTorch×GPU セットアップ手順

目次

1. はじめに
2. セットアップ
3. 動作確認

はじめに

自分用のメモとして、ディープラーニングフレームワークPyTorchをGPUで利用する際のセットアップ手順を記します。

インストール

• PyTorch・Python・CUDAそれぞれのバージョンによって相性問題があるため注意
  • 例えば3000番台のGPUであれば PyTorch 1.7 以上・Python 3.8 以上・CUDA 11.x
• AnacondaのGUIからPyTorchをインストールしない

下記リンクからインストールするコマンドを確認しておく

アーカイブから必要なバージョンのCUDA Toolkitをダウンロードしておく

PyTorch用の環境をつくって、Anaconda promptでPyTorchをインストールする

Poweshell Prompt

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

動作確認

最後にJupyter NotebookでGPUの動作確認をする