Win11 WSLでPyTorch環境構築(備忘録)

備忘録のための雑記です

WSL&Ubuntuのインストール

Microsoft Storeにアクセスし、WSLとUbuntuをインストールします。

Ubuntuを起動しユーザーネーム・パスワードを入力しセットアップ完了

Python開発環境セットアップ

VScodeのインストール

WSLパッケージのインストール

CUDA有効化

CUDA toolkitインストール
Win11用とWSL用を両方インストール
WSLの方はLinux→x86_64→WSL-Ubuntu→2.0→deb(お好みで)を順にクリック

nvidia driverもインストール

仮想環境

venvを有効化するために下記のコマンドを順に叩きます。(venvは個人的な好み)

bash

$ sudo apt-get update
$ sudo apt-get install software-properties-common
$ sudo add-apt-repository ppa:deadsnakes/ppa
$ sudo apt-get update
$ sudo apt-get install python3.X-venv <-versionは自分の環境に合わせる

仮想環境の立ち上げコマンド

bash

$ python3 -m venv env_name
$ cd env_name/ && source bin/activate && code ./

torchでCUDA認識してるかチェック

bash

$ pip install torch torchaudio torchvision

python

import torch
print(f"CUDA is available:{torch.cuda.is_available()}")

お疲れさまでした。

Reference