しんさんのブログ

科学や技術のこと読書のことなど

Windows11でWSL linux上でのCUDAの設定メモ

Windows11でWSL上のUbuntuにCUDAを設定したときのメモです

1) Ubuntu 20.4 LTSをWindowsアプリストアからインストール
2) Windows側でCUDA for WSLのGPUドライバをインストール
3) Ubuntu側でCUDAをインストール
以下のサイトから適切なパッケージを選択し
https://developer.nvidia.com/cuda-zone
Installerに従ってインストールする
今回は、CUDA 11.6をインストール

/usr/local/cuda-11.6/bin/nvcc -V

で正しくバージョンが表示されればインストール成功

pathをつなぐために、.profileに
export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
を追加
4) Ubuntu上の各種ツールインストール
このページに従って
https://docs.microsoft.com/ja-jp/windows/wsl/tutorials/gui-apps
Gedit、GIMPVLCをインストール
Linux 向け Google Chromeもインストール
5) Anacondaのインストール
以下のサイトからインストール用shをダウンロード
https://www.anaconda.com/products/individual
途中で、yes/noと聞かれるところがあるので適宜入力する。
UbuntuをリスタートするとAnaconda環境が使用できるようになる。
6) cuDNNのインストール
以下のNVIDIAサイトを参照しながらcuDNNをインストールする
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

  • WSL2上のLinuxにDockerをインストール
    上記の手順は佐々木さんの記事を参考に実行しました。
    CUDA on WSLに関しては実際は佐々木さんの記事の通りではなく、Nvidiaの以下のサイトの手順に従ってインストールしました。 CUDA Toolkit 11.6 Update 1 Downloads | NVIDIA Developer

qiita.com

また次のページの情報も役に立ちます
WSL2 上の Ubuntu での NVIDIA CUDA ツールキット, NVIDIA cuDNN, PyTorch, TensorFlow 2.7 のインストールと動作確認(Windows 上)

TensforlowのGPU版が動かないときは以下のサイトを参照 WSL2 + Ubuntu20.04 + CUDA 11.4 で TensorFlow 環境構築 (2021.08)

Google-Chromeのインストールや他のLinux on WSL2のアプリの設定は以下のブログを参考にしました。

docs.microsoft.com

WSL2のCUDAに関するNVIDIA公式ドキュメント

https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started

その他参考になるサイト