01.Pytorch安装

# 01.安装PyTorch

① Mac 终端安装 Miniconda（Apple M1/M2 适配版）

# 下载 Miniconda 安装包（适用于 M1/M2 芯片）
curl -o ~/miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh

# 运行安装脚本
bash ~/miniconda.sh

# 关闭并重新打开终端，或者手动运行：
source ~/.zshrc  # 如果使用 Zsh
source ~/.bashrc  # 如果使用 Bash

# 验证安装成功
conda --version

1
2
3
4
5
6
7
8
9
10
11
12

② 创建 PyTorch 环境

conda create -n pytorch_learn python=3.9 -y
conda activate pytorch_learn

1
2

③ 安装 PyTorch

# pytorch-nightly 是官方 MPS 支持的开发版本，能最大化支持 Apple M1 GPU
conda install pytorch torchvision torchaudio matplotlib -c pytorch-nightly

1
2

④ 测试 PyTorch 是否安装成功

import torch

# 检查 PyTorch 版本
print(torch.__version__)

# 检查是否支持 MPS（M1/M2 GPU 加速）
print(torch.backends.mps.is_available())

# 测试 Tensor 运算
x = torch.rand(3, 3).to("mps")
print(x)

1
2
3
4
5
6
7
8
9
10
11

上次更新: 2025/3/13 17:01:28

← 20.回归 02.Tensor 张量→