WSL2 上の PyTorch に GPU を認識させて深層学習環境をつくる本記事では、WSL2 上の PyTorch から GPU を使用できる深層学習用の環境を構築していきます。環境は下記の通りです。 OS : Windows 11 Home (21H2) CPU : AMD Ryzen 7 3700X 8 コア RAM : 16 GB GPU : NVIDIA GeForce RTX 2080 Ti 作業フロー GPU を利用して PyTorch で記述したモデルの学習をするまでの道のりはこんな感じ。 Windows に NVIDIA ドライバーをインストール WSL (Ubuntu) に CUDA Toolkit をインストール WSL (Ubuntu) に cuDNN をインストール WSL (Ubuntu) から GPU を認識できているか動作確認 WSL (Ubuntu) に PyTorch をインストール 用語辞典 NVDIA ドライバー NVID
