M1 MacでPyTorch2.1環境構築
概要
- M1 Macに以下のPyTorch環境を構築する
- conda(miniconda)は導入済みとする
環境
1.仮想環境を作成
$ conda create --name torch2_1 python=3.11
$ conda activate torch2_1
2.PyTorchのインストール
$ conda install pytorch::pytorch torchvision torchaudio -c pytorch
- 2023/12/31時点での実行
- これによりインストールされるのは以下のものとなる
- pytorch:2.1.2
- torchvision:0.16.2
- torchaudio:2.1.2
3.パッケージのインストール
$ conda install numpy opencv matplotlib scikit-learn pandas jupyter tqdm imgaug torchmetrics torchinfo timm ttach lightning
4.動作チェック
>>> import torch
>>> torch.backends.mps.is_available()
True