MacM1部署深度学习

844 阅读1分钟

Mac M1 部署深度学习

reference:makeoptim.com/deep-learni…

第一步:配置miniconda

首先,m1芯片不支持anaconda,因此千万不要安装anaconda,选择配置miniforge3 下载地址为:github.com/conda-forge… ,寻找适配Apple Silicon的版本

image.png

其次,按照如下代码从shell文件安装miniforge3

mv Miniforge3-MacOSX-arm64.sh ~/
Chmod +x Miniforge3-MacOSX-arm64.sh
./Miniforge3-MacOSX-arm64.sh

最后,查看安装结果

conda --version

第二步:安装apple支持的tensorflow

一句话,说白了就是在apple的channel下用pip安装包。 首先,新建tf环境,因为miniforge的原因,m1芯片仅支持3.8以上的python:

conda create -—name tf26 python=3.8
# 激活环境
conda activate python38

参照apple开发者文档developer.apple.com/metal/tenso… ,安装依赖包

conda install -c apple tensorflow-deps

当然,你可能之前安装过macos的tf2.4版本,若升级为2.6需要:

# uninstall existing tensorflow-macos and tensorflow-metal
python -m pip uninstall tensorflow-macos
python -m pip uninstall tensorflow-metal
# Upgrade tensorflow-deps
conda install -c apple tensorflow-deps --force-reinstall
# or point to specific conda environment
conda install -c apple tensorflow-deps --force-reinstall -n my_env

然后选择升级版本

#conda install -c apple tensorflow-deps==2.5.0
conda install -c apple tensorflow-deps==2.6.0

第二,正式安装macos的tensorflow

python -m pip install tensorflow-macos

最后,安装metal文件,刚开始不知道metal是什么,参照博客docs.unity3d.com/cn/2019.4/M… Metal 允许您在运行应用程序时选择 GPU 设备。因此,可使用不同的 GPU 设置来测试项目,或使用低功耗 GPU 省电。同时,Metal 是 Apple 设备的标准图形 API。Unity 支持 iOS、tvOS 和 macOS 上的 Metal(Standalone 和 Editor)。Metal 在 Apple 平台上具有比 OpenGL ES 更丰富的功能集。

python -m pip install tensorflow-metal

第三步:正式使用