本文主要介绍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://www.toymoban.com/news/detail-849994.html
https://mp.weixin.qq.com/s/EBnX-h4UqGV9KBJJ43f2tA文章来源地址https://www.toymoban.com/news/detail-849994.html
到了这里,关于MacOS M2:配置Tensorflow-GPU版的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!