首先,我们默认已经安装了conda。第一步就是安装TensorFlow在Apple上面的依赖。
conda install -c apple tensorflow-deps -y

第二步是安装TensorFlow-macOS。
python -m pip install tensorflow-macos
如果你网速不够可以使用清华源。
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-macos

第三步是安装TensorFlow-Metal。
python -m pip install tensorflow-metal
如果你网速不够可以使用清华源。
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-metal
这样子TensorFlow就成功安装在了m1芯片的MacBook Pro上了。
然后就可以开心地在m1芯片上训练模型了。

本文介绍了如何在搭载M1芯片的Mac上安装TensorFlow和TensorFlow Metal。首先确保已安装conda,接着分步骤安装TensorFlow的Apple依赖、TensorFlow-macOS,并提供了使用清华源加速下载的建议。最后,成功安装后即可在M1 MacBook Pro上进行模型训练。
最低0.47元/天 解锁文章
345

被折叠的 条评论
为什么被折叠?



