[TensorFlow] arm架构macOS安装TensorFlow并开启GPU加速

在M1系列的arm架构macOS上安装TensorFlow时,如果直接使用pip install tensorflow安装会出现如下错误:

ERROR: Could not find a version that satisfies the requirement tensorflow (from versions: none)
ERROR: No matching distribution found for tensorflow

这个错误说明了无法直接从pip上安装TensorFlow,需要一些其他办法来进行安装。除此之外,为了加快TensorFlow的执行速度,还需要安装TensorFlow Matel Plugin来启用GPU加速。

本篇文章将参考苹果官方文档

安装步骤

配置虚拟环境

如果无需虚拟环境,直接将TensorFlow安装在默认环境中,可以跳过这一部分。

因为macOS自带多个版本的Python,而Homebrew也会自带Python环境,导致会有好几个Python同时存在path中。除此之外,pip与Python也可能会出现不对应的情况,再加上pip安装的依赖会全局有效,强烈建议使用conda统一管理Python环境。

Miniforge附带了conda,conda是一个适用于Python的依赖管理与环境管理工具。可以使用Homebrew安装

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值