适用于macOS 11.0的硬件加速TensorFlow和TensorFlow插件

2020-12-06 18:27:33

该预发行版为macOS 11.0+提供了硬件加速的TensorFlow和TensorFlow插件。通过Apple的ML Compute框架,在具有M1的Mac和基于Intel的Mac上支持本机硬件加速。

可以从发行版中下载包含Python软件包和安装脚本的存档。

这将验证您的系统,要求您进行确认,然后创建一个安装了TensorFlow for macOS的虚拟环境。

或者,从发行版中下载存档文件。归档文件包含安装脚本,TensorFlow的加速版本,TensorFlow插件以及所需的依赖项。

无需对现有TensorFlow脚本进行任何更改即可将ML Compute用作TensorFlow和TensorFlow插件的后端。

有一个可选的mlcompute.set_mlc_device(device_name =’any')API,用于ML Compute设备选择。 device_name的默认值为“#”,这表示ML Compute将选择系统上最佳的可用设备,包括多GPU配置上的多个GPU。其他可用选项是“ cpu”和“ gpu”。请注意,在急切模式下,ML Compute将使用CPU。例如,要选择CPU设备,可以执行以下操作:

#导入mlcompute模块以使用可选的set_mlc_device API进行ML Compute的设备选择。来自tensorflow.python.compiler.mlcompute import mlcompute#选择CPU device.mlcompute.set_mlc_device(device_name ='cpu')#可用选项为' cpu',gpu'和'any'。