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())