TorchAudioとPyTorchを使用するプロジェクトを実行するために、WSL 2にCUDAをインストールしようとしています。ドキュメントに従って手順を正確に実行しました。ここ。
ただし、必要なコードを実行すると、次のエラーが発生します。
RuntimeError: No CUDA GPUs are available
参考までに、私は NVIDIA RTX 2060 を持っており、NVIDIA ドライバーも正しくインストールされています。これに対して推奨される次の手順や既知の解決策はありますか?
追加情報として、予想どおり最新の Windows プレビュー ビルドを使用しており、Ubuntu バージョンは 20.04 (Microsoft ストアから直接ダウンロード) です。
答え1
cuda ツールキットがインストールされていて、すべてが整っていること (WSL の nvidia-smi が動作しているなど) を前提としています。pytorch バージョンを 1.10 から 1.8.2LTS にダウングレードすると役立つ可能性があります。これは私の場合はうまくいきました。他の人がうまくいったと報告している以前の試み (私の場合はうまくいきませんでした) は、NVidia ディスプレイ ドライバーを 5xx ではなく 472.12 にダウングレードすることです。
pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html