MacbookPro 2019 安装tensorflow GPU加速版

穆栩萌霖

已于 2023-01-13 10:25:20 修改

阅读量574

点赞数

分类专栏：深度学习文章标签： tensorflow 深度学习 python

于 2023-01-10 20:59:39 首次发布

本文链接：https://blog.csdn.net/qq_38393270/article/details/128636228

版权

深度学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

苹果没有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())