Pythonアプリケーション実行時に "AssertionError: Torch not compiled with CUDA enabled" が発生する

Pythonアプリケーション実行時に "AssertionError: Torch not compiled with CUDA enabled" エラーが発生しアプリケーションが実行できない現象の対処法の紹介です。

現象

アプリケーションの実行時、特に処理時に以下のエラーが発生します。

または、

または、

原因

アプリケーションで利用してPyTorchがインストールされていない、 または、対応するCUDAのバージョンがインストールされていない可能性があります。(CPU/GPU版、バージョン違いなど)
あるいは、GPUを利用する環境で、インストールされているTorch / PyTochのバージョンがCPU専用バージョンであることが原因です。

対処法

以下の対処法1、または、対処法2のどちらかを実施します。

対処法1 : ライブラリの再インストール

CUDA Toolkit 最新版のインストール

CUDA Toolkit 最新版をインストールします。CUDA Toolkit のインストール手順はこちらの記事を参照してください。

venvディレクトリの削除

ライブラリの不一致等の可能性もあるため、venvディレクトリをすべて削除します。

アプリケーションの実行

アプリケーションを実行し、再度ライブラリをインストールします。

以上の手順で解消される場合があります。

対処法2 : PyThorchの再インストール

PyTochのアンインストール

CPU専用版のPyTochをアンインストールします。コマンドプロンプトまたは、venv環境で次のコマンドを実行します。

pip uninstall torch

キャッシュのクリア

次のコマンドを実行し、キャッシュをクリアします。

pip cache purge

PyTochのインストール

PyTochをインストールします。このとき、アプリによっては、最新版が動作環境でない場合もあるため、インストールするPyTochによって実行するコマンドが異なります。
コマンドは(https://pytorch.org/get-started/previous-versions/)にて確認できます。

PyToch 2.2系では新しすぎて動作しないアプリの場合は、PyToch 2.1系または、2.0系をインストールします。

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121

pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118

pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

著者

ジョニー

iPentecのプログラマー、最近はAIの積極的な活用にも取り組み中。
とっても恥ずかしがり。