当前位置:   article > 正文

mac开启TensorFlow GPU加速_mac no match found for: tensorflow-gpu. search: *t

mac no match found for: tensorflow-gpu. search: *tensorflow-gpu*

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

GitHub官方地址

下面给出安装方法

1.安装anaconda

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

3.解压

M1进入Arm64   Intel进入X86

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全家桶问题,未测试。

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/在线问答5/article/detail/896665
推荐阅读
相关标签
  

闽ICP备14008679号