即使安装了m1芯片的anaconda，其python仍然是intel架构的，无法利用m1的优势，而pytorch针对arm架构做了优化，因此尝试使用arm下的pytorch
; 1、将base从anaconda切换到miniforge（arm架构的conda）
参考
conda→minforge: /Users/用户名/miniforge3/bin/conda init zsh
重启终端才能生效
; 2、创建一个虚拟环境
conda create -n m1py310 python=3.10
m1py310是虚拟环境的名字，根据自己的喜好来命名
3、激活环境，安装最新的arm架构pytorch
conda activate m1py310
pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
实测速度提升了五到六倍！（ 基础的Unet）
intel python:
arm python:
再试一下mac的GPU
PS：自从买了这个电脑，GPU一直没动过，今天终于动起来了，内心os：GPU起来干活！哈哈
调用方式：
device = torch.device("mps")net.to(device)
对比：
（b，c，w，h）–>（2，1，512，512）
类别时间intel cpu50arm cpu11arm gpu4
Original: https://blog.csdn.net/qq_40494242/article/details/124945322
Author: 爱学习的pumpkin
Title: 安装arm架构的pytorch
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/709911/
转载文章受原作者版权保护。转载请注明原作者出处！