栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Mac M1的PyCharm中安装TensorFlow与创建工程的方法(支持GPU)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Mac M1的PyCharm中安装TensorFlow与创建工程的方法(支持GPU)

本文介绍在Mac M1的PyCharm中安装TensorFlow与创建工程的方法,在2021的MacBook Pro (M1 Pro处理器)验证OK。

安装TensorFlow与创建工程是在Miniforge环境中进行的。[Miniforge](GitHub - conda-forge/miniforge: A conda-forge distribution.)是一个经过包装的Conda环境,它支持Apple M1平台。最后创建的工程是在Conda虚拟环境中运行的,安装的TensorFlow只在这个项目中生效,不会影响全局环境中的包。

  1. 下载Miniforge安装脚本,在终端中执行下面的命令安装:

    chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
    sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
    
  2. 在PyCharm当中新建工程,选择用conda新建环境

    到此新工程已经创建,PyCharm的终端将自动进入conda的虚拟环境,下面的3个步骤的命令在PyCharm自带的终端中输入执行。

  3. 从Apple的仓库安装tensorflow的依赖,

    conda install -c apple tensorflow-deps
    
  4. 安装Mac OS版本的TensorFlow

    python -m pip install tensorflow-macos
    
  5. 安装tensorflow-metal,使TensorFlow支持GPU计算

    python -m pip install tensorflow-metal
    
  6. 验证安装是否成功。在PyCharm的“Python控制台”中输入如下指令查询TensorFlow的版本以及查看是否支持GPU,如果能够返回一下结果说明安装成功:

    在活动监视器中可以查看是否是GPU进行运算

    参考:
    https://developer.apple.com/metal/tensorflow-plugin/

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/445349.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号