macOS下安装运行PyTorch及速度测试

macOS下安装运行PyTorch及速度测试。

PyTorch支持多种平台，包括Linux/Windows/macOS，甚至支持Android和iOS，今天只讲macOS。

如果你的macOS上已安装Python3，直接使用pip3命令安装PyTorch即可。

pip3 install torch torchvision torchaudio

最新的1.12正式版已经支持使用MPS进行训练，对于Apple Silicon机型来说，就是可以使用GPU来训练模型，理论速度要比CPU快5~20倍。

我们使用PyTorch官方的衣服分类来做测试，下面是使用CPU和使用GPU训练的结果。

这个结果跟TensorFlow的一样，M1的CPU速度比Intel的快了不少，但GPU比CPU还慢，可能PyTorch对MPS的支持不完善。

但是相对来说，PyTorch在不同平台下的配置比TensorFlow要统一，而且训练时间上的差距也小的多，不像TensorFlow，GPU训练比CPU要慢上好几倍，真不知道是哪里出了问题，Apple所宣传的看起来秒RTX 3090独立显卡的GPU，在模型训练上居然被CPU反超了，更别提RTX系列的独立显卡了。