macOS M2 は conda を使用して pytorch 環境を構成します

1.コンダ

まず、conda が必要です。miniconda のインストールと構成のプロセスは、http://t.csdn.cn/ShcNbで確認できます。

2.公式サイトに入る

対応するバージョンを選択し、次のコマンドをコピーします

conda install pytorch torchvision torchaudio -c pytorch

3. 仮想環境の作成と pytorch のダウンロードを開始します

  • ターミナルに、入力します

conda create -n torch-gpu python=3.9
conda activate torch-gpu
conda install pytorch torchvision torchaudio -c pytorch

最初のコマンドは、torch-gpu、python バージョン 3.9 という名前の仮想環境を作成することです。

2 番目のコマンドは、環境をアクティブにすることです。

3 番目のコマンドは、pytorch をインストールすることです。

4. テスト

python
import torch
torch.__version__
torch.device('mps')

おすすめ

転載: blog.csdn.net/qyqyqyi/article/details/129714370