WSL2+Docker+GPUでPyTorchを動かす

最近GPU付きのWindowsパソコン(Windows11 Pro)を購入し、GPUをフル活用しようと色々試したので、記録して記事をあげておきます。

ただやることは簡単で以下の手順をやるだけです。(詳細なものは今後予定)

・WSL2(Ubuntu)のインストール

・WSL2上のUbuntuにこちらの手順に従って、docker+αをインストール

自分の手元の環境ではこの手順だけで対応できました。

端末によってはNVIDIAのドライバーのインストールが必要な場合もあるかもしれませんので、ご了承ください。

ちなみに私が購入したパソコンのGPUはGTX1650です。(あくまでも学習用)