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的提示

作者:jacky chen原文地址:https://segmentfault.com/a/1190000043508405

%s 个评论

要回复文章请先登录注册