タイトル通りですが、WSL2のUbuntuにCUDAをインストールして、PyTorchでCUDAを利用する方法を紹介します。 そこまで難しい設定はありませんが、2021年4月17日時点ではWindows Insider Programに参加して、Windows10のビルドバージョンを20145以上にする必要があります。 多少面倒なのとdev版のビルドバージョンを利用する必要がありますが、WSLでGPUを使えるようになるのは便利すぎます。 実行環境Windows10 Pro (OSビルド 21359)Ubuntu 20.04 on WSL2GeForce MX350CUDA Toolkit 11.3Python 3.8PyTorch 1.8.1さらっと書きましたが、GeForce MX350でCUDA&PyTorchを使おうとする人ってどのくらいいるのでしょうか、、、 Windows10のビ
![WSL2のUbuntuにCUDAをインストールしてPyTorchを動かす](https://cdn-ak-scissors.b.st-hatena.com/image/square/1ad9a4acdc2214b897a394138b072fea56f0b426/height=288;version=1;width=512/https%3A%2F%2Ftakake-blog.com%2Fwp-content%2Fuploads%2F2021%2F04%2Fwsl2-cuda-pytorch-768x403.jpg)