Apple M1安装Tensorflow

本文最后更新于：2022年7月2日 晚上

前言

最近开始在m1上跑tensorflow,但是发现一旦导入’tensorflow’这个包，就会导致整个Python退出。这个Python是由Anaconda创建的虚拟环境，排查了一下才发现原来之前安装的一直是X86的Python.

在网上看到的教程，很多人都说需要安装miniconda，或者是Python3.7 . 但是经过测试，发现是可以安装在Anaconda的Python3.9的.

开始

安装ARM的Anaconda

  1. 访问Anaconda官网 https://www.anaconda.com/
  2. 我下载的是图形安装版(64-Bit (M1) Graphical Installer)
  3. M1的Anaconda安装好后，不会自动安装Anaconda-Navigator,需要的话可以自己安装
    1
    conda install anaconda-navigator

安装Python环境

1
conda create --name py39 python=3.9

安装 Tensorflow

首先进入conda环境

1
conda activate py39

安装tensorflow依赖包

1
conda install -c apple tensorflow-deps

安装tensorflow

1
pip3 install tensorflow-macos

安装tensorflow-metal(GPU加速)

1
pip3 install tensorflow-metal

验证

代码

1
2
3
4
5
6
7
8
9
10
import sys
import tensorflow.keras
import tensorflow as tf
print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {tensorflow.keras.__version__}")
print(f"Python {sys.version}")
print(tf.config.list_physical_devices('GPU'))
gpu = len(tf.config.list_physical_devices('GPU'))>0
print("GPU is", "available" if gpu else "NOT AVAILABLE")
tf.test.is_gpu_available()

输出

1
2
3
4
5
6
7
Tensor Flow Version: 2.9.2
Keras Version: 2.9.0
Python 3.9.12 (main, Jun  1 2022, 06:34:44) 
[Clang 12.0.0 ]
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
GPU is available
True

参考

  1. 2022最新Apple M1/Mac M1 配置安装Tensorflow环境方法(亲测可用). https://zhuanlan.zhihu.com/p/474212619
  2. Apple MacBook M1 Anaconda安装 Tensorflow. https://zhuanlan.zhihu.com/p/445535362