M1 MacでPyTorch2.0環境構築

Environment

M1 MacでPyTorch2.0環境構築

概要

  • M1 MacにPyTorch2.0の環境を構築する
  • conda(miniconda)は導入済みとする

環境

1.仮想環境を作成

$ conda create --name torch2_0 python=3.9.16
$ conda activate torch2_0

※補足：Python3.10、3.11系ではパッケージの互換性の問題でインストールできず

2.PyTorchのインストール

$ conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 -c pytorch

3.パッケージのインストール

conda install numpy opencv matplotlib scikit-learn pandas jupyter tqdm

4.動作チェック

>>> import torch
>>> torch.backends.mps.is_available()
True