即使安装了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 cpu | 50 |
arm cpu | 11 |
arm gpu | 4 |