MacOS M2：配置Tensorflow-GPU版

本文主要介绍MacOS 12.x系统下的Tensorflow配置。总的来说，配置Mac版本的Tensorflow只需要三步：第一步配置一个虚拟环境，建议选择miniconda；第二步创建conda环境；第三步安装Tensorflow。

1、安装 Miniconda

1.1 彻底删除anaconda

（1）下载  anaconda-clean   用于删除相关配置文件

##  在终端中输入：
conda install anaconda-clean
anaconda-clean --yes

2）删除anaconda、miniconda、miniforge所在文件夹

##  在终端中输入：
rm -rf ～/anaconda3
rm -rf ～/opt/anaconda3
rm -rf ～/miniforge3
rm -rf ～/miniconda3
rm -rf .conda
rm -rf .condarc

3）打开 .bash_profile  ，删除anaconda路径

#  vim打开环境配置
vim .bash_profile
#  运行后找到下面这行代码，在代码前输入dd删除
#  export PATH="/Users/your_username/anaconda3/bin:$PATH"

1.2 安装Miniconda

去miniconda官网选择下载，默认路径。https://docs.conda.io/en/latest/miniconda.html 

 

 

2、配置环境

2.1 创建虚拟环境

在Miniconda3中创建环境，存放在Miniconda3的env文件夹中

# 在终端中输入：
conda create -n ***** python=3.9
# *****为你给这个配置环境的命名

2.2 激活虚拟环境

#  在终端中输入：
conda activate *****
#  然后关闭终端，重新打开终端

2.3 miniconda相关操作代码

查看当前系统下的环境：conda info -e
查看conda版本：conda -V
查看所有已经安装的包：conda list
创建新的虚拟环境：conda create -n **** python=3.9   （****为虚拟环境名称）
安装包：  conda install ****      （****为包名称）
切换虚拟环境：conda activate ****   （****为虚拟环境名称）
退出环境：conda deactivate
删除某个虚拟环境：conda remove -n **** --all  （****为虚拟环境名称)

3、配置Tensorflow-macos

3.1 安装依赖包 tensorflow-deps 

如果不装这个包，后续将无法成功安装的Tensorflow包

conda install -c apple tensorflow-deps

3.2 配置 Tensorflow 及 jupyter notebook

注意：这一步以及后续的packages都是在虚拟环境 ***** 当中安装

conda activate *****
##  *****为你给这个配置环境的命名
###  1.安装 tensorflow-macos，指定版本2.9，新版本暂时不能调用GPU
python -m pip install tensorflow-macos==2.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

##  若下载出错则需要先下载torch，之后重新下载tensorflow-macos
#  pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple/

###  2.安装Apple官方维护的 tensorflow-metal
python -m pip install tensorflow-metal==0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

##  3.安装jupyter等工具包
conda install jupyter jupyterlab pandas matplotlib numpy

4、测试 Tensorflow-macos

4.1 新版本packages的bug及解决办法

刚开始我是照着Apple官网安装了最新的packages，后来发现无法调用GPU加速而一直报错，但是使用CPU的话程序正常运行，说明问题出在packages上：

 tensorflow/core/framework/op_kernel.cc:1830] OP_REQUIRES failed at xla_ops.cc:418 : NOT_FOUND: could not find registered platform with id: 

最后在Apple的反馈平台中找到了答案：装回旧版包 

tensorflow-macos==2.9 ; 

tensorflow-metal==0.5.0 ; 

4.2 运行 Tensorflow-macos

import numpy as np
import tensorflow as tf

x = np.random.random((10000, 5))
y = np.random.random((10000, 2))
x2 = np.random.random((2000, 5)
y2 = np.random.random((2000, 2))

inp = tf.keras.layers.Input(shape = (5,))
l1 = tf.keras.layers.Dense(256, activation = 'sigmoid')(inp)
l1 = tf.keras.layers.Dense(256, activation = 'sigmoid')(l1)
l1 = tf.keras.layers.Dense(256, activation = 'sigmoid')(l1)
l1 = tf.keras.layers.Dense(256, activation = 'sigmoid')(l1)
l1 = tf.keras.layers.Dense(256, activation = 'sigmoid')(l1)
o = tf.keras.layers.Dense(2, activation = 'sigmoid')(l1)
model = tf.keras.models.Model(inputs = [inp], outputs = [o])
model.compile(optimizer = "Adam", loss = "mse")

参考文献

[1] Apple官方-Tensorflow安装教程

https://developer.apple.com/metal/tensorflow-plugin/

[2] Apple Developer Forums - tensorflow-metal 

https://developer.apple.com/forums/tags/tensorflow-metal

[3] 【客官投稿】MAC M1芯片上如何快速安装TensorFlow？

https://mp.weixin.qq.com/s/EBnX-h4UqGV9KBJJ43f2tA