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训练的结果。

机器类型 时间
Mac mini 2018 Intel CPU 48s
Mac mini 2020 M1 CPU 12s
Mac mini 2020 M1 GPU 18s

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

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

Leave a Comment

豫ICP备19001387号-1