Cuda on WSL2 を有効にする - Himeyama/himeyama GitHub Wiki
Windows 10 / 11 version 21H2 のインストール
GPU ドライバーのインストール
既存の CUDA 機械学習のワークフローを有効にするために、https://developer.nvidia.com/cuda/wsl をダウンロードしてインストールする。
WSL2 のセットアップ
WSL2 を有効にし、glibc ベースのディストリビューション (例えば Ubuntu や debian) をインストールする。 設定の Windows Update を選択し、アップデートのチェックを行い最新のカーネルにする。
これらの機能のために、5.10.43.3 以上のバージョンが必要である。 PowerShell によって確認を行う。
> wsl cat /proc/version
Linux version 5.10.60.1-microsoft-standard-WSL2 (oe-user@oe-host) (x86_64-msft-linux-gcc (GCC) 9.3.0, GNU ld (GNU Binutils) 2.34.0.20200220) #1 SMP Wed Aug 25 23:20:18 UTC 2021
NVIDIA Docker を経由して既存の Linux ワークフローや WSL を含む PyTorch または TensorFlow のインストールを 使用し始めることができる。
wsl の NVIDIA ドライバーは不要
nvidia-smi
を実行するとエラーが出るので、nvtop
をインストールする。
sudo apt update -y
sudo apt install nvtop -y
CUDA Toolkit のインストール
https://developer.nvidia.com/cuda-toolkit-archive
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
# sudo apt-key add 7fa2af80.pub
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
sudo apt-get update
sudo apt-get install -y cuda-toolkit-11-8
参考
https://docs.microsoft.com/ja-jp/windows/ai/directml/gpu-cuda-in-wsl