苹果M1支持TensorFlow 2.5

3,700

开始使用tensorflow-metal PluggableDevice

在你的Mac上用TensorFlow加速机器学习模型的训练。

安装TensorFlow v2.5和tensorflow-metal PluggableDevice,在Mac的GPU上用Metal加速训练。你可以在这里了解更多关于TensorFlow PluggableDevice的信息。

操作系统要求

  • macOS 12.0以上

目前不支持

  • 多GPU支持
  • 英特尔GPU的加速
  • V1 TensorFlow网络

安装说明

环境设置

x86 : AMD

(可选)Create venv: python3 -m venv ~/tensorflow-metal

source ~/tensorflow-metal/bin/activate
python -m pip install -U pip

arm64 : M1

下载并安装Conda环境

chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
source ~/miniforge3/bin/activate

安装Tensorflow依赖项。

conda install -c apple tensorflow-deps

安装基础tensorflow。

python -m pip install tensorflow-macos

安装金属插件。

python -m pip install tensorflow-metal

原文链接:developer.apple.com/metal/tenso…