https://learn.microsoft.com/ja-jp/windows/ai/directml/gpu-cuda-in-wsl
nvidia-smiエラーなく、対象のGPUが表示されればOK。 表示されていない場合は、NVIDIAのページよりドライバをダウンロード&インストールする。 ※メーカーPCの場合は、メーカーのページより適宜ダウンロード&インストールする。
下記 URL より、対象 OS 等選択すると、手順が表示される。
CUDA Toolkit Installer の手順に従って インストールする。
※ Driver Installer の方は、インストールしないこと。
https://developer.nvidia.com/cuda-downloads
※ wget がない場合は、事前にインストールしておく。
sudo apt install wget
sudo yum install wget
途中、うまくいかない場合は、以下確認。 WSL 上で、下記実行し表示されれば認識はしている。
nvidia-smi
cuda のバージョン確認は、下記コマンドで可能
/usr/local/cuda/bin/nvcc -V
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
★最新のコマンドは次の PyTorch のページで確認ください。
※Install PyTorch にて、自分の環境を選択して、Run this Command: に表示されるコマンドを実行すればOKです。
python -c "import torch; print(torch.cuda.is_available())"