乐趣区

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

退出移动版