macos在m1芯片下安装tensorflow2

网上搜索了大量文章后,将安装过程写下来,也给予其他有同样需求的伙伴一些方法。

  1. 安装miniforge
    下载地址: Miniforge3-4.10.3-6-MacOSX-arm64.sh
# 先进入miniforge下载的路径中
bash Miniforge3-4.10.3-6-MacOSX-arm64.sh

安装好后输入conda,没有报错就表明安装成功了。
注:anaconda和mini forge可以同时安装,需要进哪些环境,就激活哪个环境的conda。

  1. 安装依赖
    建议新建一个虚拟环境安装tensorflow,而不要直接在base中操作,避免出现问题无法重置。
# 创建虚拟环境
conda create -n 虚拟环境名 python=3.8
# 进入该虚拟环境
conda activate 虚拟环境名

进入虚拟环境后再继续安装。

conda install -c apple tensorflow-deps
  1. 安装base tensorflow
python -m pip install tensorflow-macos

以上安装都成功后,说明你的tensorflow就安装成功啦~
在终端中最后查看下tensorflow是否真正安装成功。

# 进入python命令行
python
# 查看tensorflow版本
import tensorflow as tf
tf.__version__

如果出现tensorflow的版本号,那么恭喜你安装成功~
特别注意:不要安装官方的文档安装metal plugin,安装成功后代码无法运行

如果想要将这个虚拟环境添加到jupyter,则需要进行以下操作。

conda install ipykernel
python -m ipykernel install --user --name 虚拟环境名 --display-name 虚拟环境名

之后再输入jupyter labjupyter notebook,如果不行,就按command+空格,输入jupyter-lab,点进去后会直接打开jupyter页面,你创建的kernel在jupyter中可以直接找到。


参考:
重磅!!!TensorFlow正式可以使用M1加速了!
Getting Started with tensorflow-metal PluggableDevice

你可能感兴趣的:(软件安装,macos,tensorflow,python)