mac开启TensorFlow GPU加速

众所周知,mac在与Nvidia决裂之后,已经不再支持cuda,让mac用户苦不堪言。2021年苹果正式发布了TensorFlow Mac优化版的Alpha版,目前已经到Alpha0.3,同时支持M1和Intel

GitHub官方地址

下面给出安装方法

1.安装anaconda

2.进入github,release下载红色箭头文件

mac开启TensorFlow GPU加速_第1张图片

3.解压

M1进入Arm64   Intel进入X86

mac开启TensorFlow GPU加速_第2张图片

4.conda创建虚拟环境,进入

依次本地安装下列包(随着版本变化,文件名可能变化,pip install --upgrade --no-dependencies --force  你的文件)

pip install --force pip==20.2.4 wheel setuptools cached-property six

pip install --upgrade --no-dependencies --force numpy-1.18.5-cp38-cp38-macosx_11_0_arm64.whl grpcio-1.33.2-cp38-cp38-macosx_11_0_arm64.whl h5py-2.10.0-cp38-cp38-macosx_11_0_arm64.whl tensorflow_addons-0.11.2+mlcompute-cp38-cp38-macosx_11_0_arm64.whl pip install absl-py astunparse flatbuffers gast google_pasta keras_preprocessing opt_einsum protobuf tensorflow_estimator termcolor typing_extensions wrapt wheel tensorboard typeguard

pip install --upgrade --force --no-dependencies tensorflow_macos-0.1a3-cp38-cp38-macosx_11_0_arm64

5.完成,进入ide,进入你创建的虚拟环境

实测,在CPU下一个epoch大概15min

在Mac-optimized TensorFlow下7min,整整快了一倍

M1下可能会比CPU快10倍,我没有M1设备

tips:我的电脑Hackintosh,配置Ryzen3500X+RX580,如果你是amd的Hackintosh用户可能运行时会遇到CPU 0 is not spport,解决方法是开启AVX指令集地址,同时可能会解决Adebo全家桶问题,未测试。

 

你可能感兴趣的:(TensorFlow,tensorflow,机器学习)