MacbookPro 2019 安装tensorflow GPU加速版

苹果没有N卡但有AMD,用苹果系统优雅的玩tensorflow就需要利用起来这块显卡
参考网站:
https://developer.apple.com/metal/tensorflow-plugin/
注意:安装前卸载tensorflow,没安装不用管
指定版本:
tensorflow-macos:2.9.0
tensorflow-metal:0.5.0
MacOS:12+

1.命令行安装环境

bash ~/miniconda.sh -b -p $HOME/miniconda
source ~/miniconda/bin/activate
conda install -c apple tensorflow-deps

2.安装TensorFlow

python -m pip install tensorflow-macos==2.9.0

3.安装tensorflow-metal插件

python -m pip install tensorflow-metal==0.5.0

测试是否能利用上GPU:运行脚本后出现“GPU:0”字样就是成功了

import os
import tensorflow as tf
from tensorflow.python.client import device_lib

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
print("GPU: ", tf.config.list_physical_devices('GPU'))
print(device_lib.list_local_devices())

在这里插入图片描述