开始使用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