CUDA環境構築で役立つリンクまとめ（WSL2・Ubuntu・Windows）

Last updated at 2024-02-24Posted at 2024-02-24

概要

環境構築のたびにやり方を調べ直しているので、環境構築の際によく見るページをまとめ、適宜注釈をつけた。
（この記事自体では環境構築の詳しい解説は行わない）

筆者がよく扱っている、Nvidia Driver、CUDA Toolkit、cuDNN、PyTorchのインストールに関してまとめた。
TensorRTについても余裕があれば後ほどまとめたい。

環境構築手順については、本記事を執筆時にWSL2（Ubuntu22.04、Ubuntu20.04）で手順を確認した。

環境構築を始める前に、以下の互換性確認用のページにまとめたページを参照しつつ、インストールするCUDA Toolkit、PyTorchのバージョンを決める
- Nvidia Driver、cuDNNのバージョンはCUDA Toolkitによって決まる
- 現実的には、使うハードウェア（GPU）の性能や使いたいライブラリ等の要件からCUDA Toolkit、PyTorchのバージョンが決まってくると思う

Nvidia Driver：Windowsに入れたものを使うのでUbuntuへのインストールは不要（Windowsの手順参照）
CUDA Toolkit・cuDNN：Ubuntuの記事1の手順でOKだが少し注意点あり
- wsl-ubuntuのリポジトリにはcuda-toolkitはあるがcuDNNが無い
  - 対策1：wsl-ubuntuのリポジトリからcuda-toolkitをインストールした後、ubuntu2204/ubuntu2004のリポジトリからcudnnをインストールする（筆者はこちらを実施、今のところ問題ない）
  - 対策2：ubuntu2204/ubuntu2004のリポジトリからcuda-toolkit、cudnnの両方をインストールする（未実施）
- CUDA Toolkit最新版以外のインストール時には、sudo apt install cuda-toolkit-12-1のように必ずバージョン指定する

GPUの世代ごとのアーキテクチャ名
- Nvidia - 比較 GeForce グラフィックスカード
GPUの世代ごとのNvidia Driver、CUDA Toolkitバージョン要件
- Nvidia - GPU Management and Deployment（5. Frequently Asked Questions - "Which GPUs are supported by the driver ?"に記載）
GPUごとのCompute Capabilityの値
- Your GPU Compute Capability
- Compute Capabilityについては以下のページがわかりやすい
  - [CUDA] NVIDIA GPUやCUDA周りの互換性を理解したかった
CUDA ToolkitのバージョンごとのNvidia Driverの要件
- NVIDIA CUDA Toolkit Release Notes（Table 2・Table 3 に記載）
cuDNNのバージョンごとのCompute Capability、Nvidia Driver、CUDA Toolkitの要件
- NVIDIA cuDNN Documentation
  - 各バージョンのcuDNN Support Matrixのページで確認
PyTorchのバージョンごとのCUDA Toolkitバージョン要件
- 下記インストール方法のページで確認
  - PyTorch - START LOCALLY（最新版）
  - INSTALLING PREVIOUS VERSIONS OF PYTORCH（過去のバージョン）

Nvidia - CUDA Toolkit Archive
- 古いバージョンだとここで出てくる手順では上手くいかないことがあるので注意（最後のsudo apt install cudaを、sudo apt install cuda-toolkit-12-1のようにcuda-toolkitのみにしてバージョン指定すれば問題ないことが多い）

CUDA Toolkitのアンインストール
- WSL2 Ubuntu 22.04 LTS の CUDA をバージョンアップする
  - アンインストール部分がわかりやすい
複数バージョンのCUDA Toolkitの切り替え
- Ubuntuでの CUDAのバージョンの切り替え方