指标

应用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的提醒