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