Pytorch – GPU と対応するドライバ、CUDA、CuDNN のバージョン

概要
使用している Nvdia GPU に対応した Driver、CUDA、CuDNN のバージョンの選び方について解説します。(2023/6/18 更新)
Pytorch を利用する場合の ドライバ、CUDA、CuDNN のバージョン選択まとめ (2023/6/1 現在)
pip
でインストールされる Pytorch が対応する CUDA のバージョンは、 11.7 です。
ドライバや CuDNN は現時点の最新のバージョンを入れて構いません。
- バージョンが 515.48.07 以上のドライバをインストールする。(CUDA 11.7 Update 1) が要求する最小のドライババージョンが 515.48.07 であるため)
- pip で Pytorch をインストールする。
pip install torch torchvision torchaudio
- CUDA 11.7 をインストールする。
- CuDNN 8 をインストールする。
Pytorch をインストールしたら、以下のコマンドで CUDA が利用可能であるか確認できます。 False になる場合は、ドライバ、CUDA、CuDNN が適切にインストールされているか確認しましょう。
import torch
print(torch.cuda.is_available()) # True になれば、CUDA が利用できる
True
Compute Capability
Compute Capability とは、利用可能な GPU の計算能力を意味し、GPU の種類によって異なります。GPU の Compute Capability は CUDA GPUs | NVIDIA Developer で確認できます。例えば、GeForce GTX 1080 の場合、「CUDA-Enabled GeForce and TITAN Products」をクリックすると、Compute Capability 6.1 であるとわかります。
GPU とドライバのバージョンの対応関係
使用している GPU に対応しているドライバをインストールする必要があります。最新のドライバは 公式ドライバー | NVIDIA で検索することで確認できます。GPU に対応しているドライバをインストールしないと適切に動作しません。
GPU の世代 | Compute Capability | ドライバのバージョン | GPU の例 |
---|---|---|---|
Ampere | 8.x | 450.36.06 以上 | GeForce RTX 3080 |
Turing | 7.5 | 418.40.04 以上 | GeForce RTX 2080 |
Volta | 7.x | 418.40.04 以上 | Tesla V100 |
Pascal | 6.x | 418.40.04 以上 | GeForce RTX 1080 |
Maxwell | 5.x | 418.40.04 以上 | GeForce GTX 980 |
Kepler | 3.x | 440.33.01 ~ 470.57.02 | GeForce GTX 680 |
Ubuntu のインストール方法
いくつか方法がありますが、Nvidia が提供する Personal Package Archive (PPA) から apt でインストールする方法を紹介します。GPU の種類は ubuntu-drivers devices
で確認できます。
古いドライバがインストールされている場合は削除します。
sudo apt remove nvidia-*
Nvidia の PPA を追加します。
sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update
パッケージの一覧を調べます。
sudo apt-cache search nvidia-driver
PPA から 公式ドライバー | NVIDIA で確認したドライバのバージョンのパッケージをインストールします。パッケージ名は適宜変更してください。
sudo apt install -y nvidia-driver-510
CUDA とドライバのバージョンの対応関係
CUDA のバージョンによって、必要なドライバのバージョンが異なります。最新の情報は Release Notes :: CUDA Toolkit Documentation の「CUDA Toolkit and Corresponding Driver Versions」の欄を参照してください。以下に 2022/2 時点の情報を記載します。例えば、Linux で CUDA 10.2.89 を使いたい場合、ドライバのバージョンは 440.33 以上である必要があります。
CUDA のバージョン | ドライバのバージョン (Linux) | ドライバのバージョン (Windows) |
---|---|---|
CUDA 12.1 Update 1 | >=530.30.02 | >=531.14 |
CUDA 12.1 GA | >=530.30.02 | >=531.14 |
CUDA 12.0 Update 1 | >=525.85.12 | >=528.33 |
CUDA 12.0 GA | >=525.60.13 | >=527.41 |
CUDA 11.8 GA | >=520.61.05 | >=520.06 |
CUDA 11.7 Update 1 | >=515.48.07 | >=516.31 |
CUDA 11.7 GA | >=515.43.04 | >=516.01 |
CUDA 11.6 Update 2 | >=510.47.03 | >=511.65 |
CUDA 11.6 Update 1 | >=510.47.03 | >=511.65 |
CUDA 11.6 GA | >=510.39.01 | >=511.23 |
CUDA 11.5 Update 2 | >=495.29.05 | >=496.13 |
CUDA 11.5 Update 1 | >=495.29.05 | >=496.13 |
CUDA 11.5 GA | >=495.29.05 | >=496.04 |
CUDA 11.4 Update 4 | >=470.82.01 | >=472.50 |
CUDA 11.4 Update 3 | >=470.82.01 | >=472.50 |
CUDA 11.4 Update 2 | >=470.57.02 | >=471.41 |
CUDA 11.4 Update 1 | >=470.57.02 | >=471.41 |
CUDA 11.4.0 GA | >=470.42.01 | >=471.11 |
CUDA 11.3.1 Update 1 | >=465.19.01 | >=465.89 |
CUDA 11.3.0 GA | >=465.19.01 | >=465.89 |
CUDA 11.2.2 Update 2 | >=460.32.03 | >=461.33 |
CUDA 11.2.1 Update 1 | >=460.32.03 | >=461.09 |
CUDA 11.2.0 GA | >=460.27.03 | >=460.82 |
CUDA 11.1.1 Update 1 | >=455.32 | >=456.81 |
CUDA 11.1 GA | >=455.23 | >=456.38 |
CUDA 11.0.3 Update 1 | >= 450.51.06 | >= 451.82 |
CUDA 11.0.2 GA | >= 450.51.05 | >= 451.48 |
CUDA 11.0.1 RC | >= 450.36.06 | >= 451.22 |
CUDA 10.2.89 | >= 440.33 | >= 441.22 |
CUDA 10.1 (10.1.105 general release, and updates) | >= 418.39 | >= 418.96 |
CUDA 10.0.130 | >= 410.48 | >= 411.31 |
CUDA 9.2 (9.2.148 Update 1) | >= 396.37 | >= 398.26 |
CUDA 9.2 (9.2.88) | >= 396.26 | >= 397.44 |
CUDA 9.1 (9.1.85) | >= 390.46 | >= 391.29 |
CUDA 9.0 (9.0.76) | >= 384.81 | >= 385.54 |
CUDA 8.0 (8.0.61 GA2) | >= 375.26 | >= 376.51 |
CUDA 8.0 (8.0.44) | >= 367.48 | >= 369.30 |
CUDA 7.5 (7.5.16) | >= 352.31 | >= 353.66 |
CUDA 7.0 (7.0.28) | >= 346.46 | >= 347.62 |
CuDNN
NVIDIA cuDNN は GPU を利用して Deep Learning に関連する計算を高速に行うためのライブラリです。対応する Compute Capability は 3.0 以上なので、昨今の GPU は対応しています。
Pytorch のバージョン
PyTorch でインストールする Pytorch のバージョンによって、対応する CUDA のバージョンが固定されます。 2022/2 現在、CUDA 11.7 と CUDA 11.8 のバージョンに対応したバイナリが配布されています。過去のバージョンは Previous PyTorch Versions | PyTorch から探せます。
-
前の記事
nvidia-smi で GPU の使用状況を確認する方法 2021.12.16
-
次の記事
記事がありません