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+

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

import tensorflow as tf


devices = tf.config.list_physical_devices()
print(devices)

你可能感兴趣的:(深度学习,tensorflow,深度学习,python)