关于m1: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 的提醒