赞
踩
众所周知,mac在与Nvidia决裂之后,已经不再支持cuda,让mac用户苦不堪言。2021年苹果正式发布了TensorFlow Mac优化版的Alpha版,目前已经到Alpha0.3,同时支持M1和Intel
下面给出安装方法
M1进入Arm64 Intel进入X86
依次本地安装下列包(随着版本变化,文件名可能变化,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
实测,在CPU下一个epoch大概15min
在Mac-optimized TensorFlow下7min,整整快了一倍
M1下可能会比CPU快10倍,我没有M1设备
tips:我的电脑Hackintosh,配置Ryzen3500X+RX580,如果你是amd的Hackintosh用户可能运行时会遇到CPU 0 is not spport,解决方法是开启AVX指令集地址,同时可能会解决Adebo全家桶问题,未测试。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。