MacM1在PyCharm中安装(支持GPU)TensorFlow方法

合集下载
  1. 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
  2. 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
  3. 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。

MacM1在PyCharm中安装(⽀持GPU)TensorFlow⽅法
本⽂介绍在Mac M1的PyCharm中安装TensorFlow与创建⼯程的⽅法,在2021的MacBook Pro (M1 Max处理器)验证OK。

安装TensorFlow与创建⼯程是在Miniforge环境中进⾏的。

[Miniforge](GitHub - conda-forge/miniforge: A conda-forge distribution.)是⼀个经过包装的Conda环境,它⽀持Apple M1平台。

最后创建的⼯程是在Conda虚拟环境中运⾏的,安装的TensorFlow只在这个项⽬中⽣效,不会影响全局环境中的包。

⼀、下载Miniforge安装脚本,在终端中执⾏下⾯的命令安装:
chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
⼆、在PyCharm当中新建⼯程,选择⽤conda新建环境
到此新⼯程已经创建,PyCharm的终端将⾃动进⼊conda的虚拟环境,下⾯的3个步骤的命令在PyCharm⾃带的终端中输⼊执⾏。

三、从Apple的仓库安装tensorflow的依赖
conda install -c apple tensorflow-deps
四、安装Mac OS版本的TensorFlow
python -m pip install tensorflow-macos
五、安装tensorflow-metal,使TensorFlow⽀持GPU计算
python -m pip install tensorflow-metal
六、验证安装是否成功。

在PyCharm的“Python控制台”中输⼊如下指令查询TensorFlow的版本以及查看是否⽀持GPU,如果能够返回⼀下结果说明安装成功:在活动监视器中可以查看是否是GPU进⾏运算。

相关文档
最新文档