CUDA on WSL2の速度比較と環境構築 - Qiita

この記事は「【マイスター・ギルド】本物のAdvent Calendar 2021」9日目の記事です。 CUDA on WSL2で機械学習環境を構築した場合に速度を落とさずに学習できるのかが気になったので、調べてみました。機械学習環境を構築する際の助けになれば幸いです。 はじめに 本記事では、CUDA on WSL2環境で機械学習をする場合と、Ubuntu直接環境(Ubuntuを直接インストールした環境)で機械学習をする場合の速度比較を行います。 また、その前段階として。2021年11月時点でのCUDA on WSL2の環境構築手順も記載します。 以下の順序で記載します。 WSL2のインストール Windows11にアップグレード CUDA導入 Pytorch環境構築 速度比較 速度比較の結果だけ見たいんだ！という方は、5. 速度比較まで飛んでください。 開発環境 CPU：Intel Cor