Windows/wsl

Windows の WSL で GPU (GeForce) を使う。

https://learn.microsoft.com/ja-jp/windows/ai/directml/gpu-cuda-in-wsl

Windows へのドライバインストール

  1. ドライバの確認 Power Shell で、次のコマンドを入力する。
    nvidia-smi
    エラーなく、対象のGPUが表示されればOK。 表示されていない場合は、NVIDIAのページよりドライバをダウンロード&インストールする。 ※メーカーPCの場合は、メーカーのページより適宜ダウンロード&インストールする。

WSL への cuda インストール

下記 URL より、対象 OS 等選択すると、手順が表示される。

CUDA Toolkit Installer の手順に従って インストールする。

※ Driver Installer の方は、インストールしないこと。

https://developer.nvidia.com/cuda-downloads

※ wget がない場合は、事前にインストールしておく。

確認

途中、うまくいかない場合は、以下確認。 WSL 上で、下記実行し表示されれば認識はしている。

nvidia-smi

cuda のバージョン確認は、下記コマンドで可能

/usr/local/cuda/bin/nvcc -V

上記までできれば、せっかくなので、PyTorch を入れる。

★最新のコマンドは次の PyTorch のページで確認ください。

https://pytorch.org/

※Install PyTorch にて、自分の環境を選択して、Run this Command: に表示されるコマンドを実行すればOKです。

トップ   一覧 検索 最終更新   ヘルプ   最終更新のRSS