m1 macos13 使用TensorFlow 问题与解决方案

目标

使用m1芯片 的mac pro进行gnn的学习时候,奇慢无比,每1000次训练在500秒以上,原因是tf没有使用GPU加速;

安装

使用miniconda,默认的conda不支持macos,无法安装mac-deps;

1. 下载安装包
   https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-...

2. 安装miniconda

   bash ./Miniconda3-latest-MacOSX-arm64.sh -b -p $HOME/miniconda 

3. 启用默认环境

   source ~/miniconda/bin/activate

4. 安装macos依赖

   conda install -c apple tensorflow-deps

5. 根据macos版本安装 tensorflow-macos 2.9版本

    SYSTEM_VERSION_COMPAT=0 pip install tensorflow-macos==2.9

6. 根据macos版本安装 tensorflow-metal 0.5.0版本

    SYSTEM_VERSION_COMPAT=0 pip install tensorflow-metal==0.5.0

   PS: 版本参照 https://developer.apple.com/metal/tensorflow-plugin/

效果

学习速度明显加快了, 会有使用gpu的提示