PyTorch 1.10がインストールできない場合は、この記事をご覧ください。
インストールできない原因と、その解決方法を説明しています。
本記事の内容
- PyTorch 1.10
- CUDAのバージョン
- インストールコマンド
それでは、上記に沿って解説していきます。
PyTorch 1.10
2021年10月22日にPyTorch 1.10.0がリリースされました。
早速、次のインストール支援ツールにも反映されています。
https://pytorch.org/get-started/locally/
PyTorch 1.10の主な改善点は、CUDA周辺です。
GPU版のPyTorchでの改善が主なポイントになるということでしょう。
GPU版利用時のCPUのオーバーヘッドが、削減されているようです。
つまり、パフォーマンス改善ということになります。
ということで、早速GPU版をインストールしようとしました。
しかし、GPU版のインストールに失敗しました。
その際、原因は大きく二つ考えられます。
- CUDAのバージョン
- インストールコマンド
それぞれを下記で説明していきます。
CUDAのバージョン
PyTorch 1.10では、次の二つのCUDAバージョンが利用可能となっています。
- CUDA 10.2
- CUDA 11.3
まずは、これらのバージョンがインストール済みかどうかを確認する必要があります。
複数の確認方法がありますが、以下が最も確実でしょう。
nvcc -V
このコマンドを実行した結果を確認します。
検証に使った環境では、次の結果です。
>nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Mon_May__3_19:41:42_Pacific_Daylight_Time_2021 Cuda compilation tools, release 11.3, V11.3.109 Build cuda_11.3.r11.3/compiler.29920130_0
CUDA 11.3を確認できています。
問題ありませんね。
CUDAのバージョンに問題がなければ、次はインストールコマンドを確認します。
インストールコマンド
公式のインストール支援ツールを利用します。
環境情報を選択すると、インストールコマンドが表示されます。
コマンドを抽出したモノが、以下。
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
このコマンドをそのまま実行すると、エラーが発生します。
エラー内容は、次のような内容です。
ERROR: Could not find a version that satisfies the requirement torchaudio===0.10.0+cu113 (from versions: 0.7.1, 0.7.2, 0.8.0, 0.8.1, 0.9.0, 0.9.1, 0.10.0) ERROR: No matching distribution found for torchaudio===0.10.0+cu113
「torchaudio 0.10.0+cu113」というパッケージは、存在しないという内容になります。
存在するのは、「0.7.1, 0.7.2, 0.8.0, 0.8.1, 0.9.0, 0.9.1,0.10.0」だけということです。
では、ここで過去バージョンのインストールコマンドを確認してみましょう。
この画像は、次の記事より取得しています。
「torchaudio===0.7.2」となっています。
「+cu110」という文字は見当たりません。
ということで、おそらくこれは支援ツールのバグです。
「torchaudio 0.10.0」とすべきなのに、「+cu113」を余計に追記しています。
以下は、「+cu113」を除外したPyTorch 1.10のインストールコマンドです。
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0 -f https://download.pytorch.org/whl/cu113/torch_stable.html
上記を実行すると、インストールが正常に始まります。
時間は結構かかるので、注意してください。
この時点でインストール済みパッケージを確認した結果は、以下。
>pip list Package Version ----------------- ------------ numpy 1.21.3 Pillow 8.4.0 pip 21.3.1 setuptools 58.3.0 torch 1.10.0+cu113 torchaudio 0.10.0 torchvision 0.11.1+cu113 typing-extensions 3.10.0.2
インストールは、問題なくできています。
よって、PyTorch 1.10がインストールできない場合は、インストールコマンドを疑ってみましょう。
追記 2021年11月18日
どうやら、「torchaudio 0.10.0+cu113」が対応されたようです。
エラーになっていたインストールコマンドが、変更なしで動くようになっています。
