【備忘録】torchとCUDAのバージョンが合わずにGPU認識しない

対処法

torchをアンインストール

pip uninstall torch torchvision torchaudio -y

指定のCUDA 11.8対応バージョン（2.4系列）をインストール
※公式サイトを参照

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

以上

キャッシュのエラーが出た場合はキャッシュ削除すればうまくいく、ときもある。

pip cache purge

P.S.　早くCUDAドライバ更新せねば。。。