安装arm架构的pytorch

即使安装了m1芯片的anaconda，其python仍然是intel架构的，无法利用m1的优势，而pytorch针对arm架构做了优化，因此尝试使用arm下的pytorch

参考
conda→minforge: /Users/用户名/miniforge3/bin/conda init zsh
重启终端才能生效

conda create -n m1py310 python=3.10
m1py310是虚拟环境的名字，根据自己的喜好来命名

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）