1.cudaをインストールする
- コマンド ラインに nvidia-smi と入力してドライバー情報を表示します。
公式 Web サイトから、対応するバージョンのインストール プログラム CUDA Toolkit Archive | NVIDIA Developer をダウンロードします。
インストールが完了したら確認します。
確認: nvcc --version と入力して確認します。
2.cudnnをインストールする
Cudann をダウンロードします: cuDNN ダウンロード | NVIDIA 開発者 (最初にアカウントを登録する必要があります)
圧縮パッケージを CUDA インストール パス (D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1) に解凍します
検証:
d:\ と入力しますProgram Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite は、
bandwidthTest.exe を実行し
、出力結果は Result = PASS となり、インストール
は成功です。
3. GPU バージョンの Pytorch をインストールします
pytroch 公式 Web サイト: https://pytorch.org/get-started/previous-versions/ にアクセスして
上記の対応するバージョンを見つけ、対応するコマンドを直接コピーしてダウンロードします。
Wheel コマンドを使用した方が明らかに高速です。