M1 mac安装PyTorch的实现步骤

 更新时间:2021年08月18日 16:28:57   作者:deephub  
本文将介绍如何在M1机器上本地安装和运行PyTorch。你使用的M1机型(Air、Pro、Mini或iMac)没有区别。感兴趣的可以了解一下

M1 macbook已经不是什么新产品了。TensorFlow官方已经给出了安装指南和效率评测。

本文将介绍如何在M1机器上本地安装和运行PyTorch。你使用的M1机型(Air、Pro、Mini或iMac)没有区别。

第一步 -安装和配置Miniforge

我花了很多时间为数据科学需求配置我的M1 Mac。但是都不能完美的解决我的问题。直到我找到了这个。根据网速的不同,完全设置需要5到10分钟。

首先,你需要安装Homebrew。它是Mac上的一个包管理器,你可以在终端上执行以下命令来安装它:

/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

如果你正在安装一个新的M1 Mac,很可能系统里面没有包含XCode构建工具,但是miniforge需要他的一些文件。所以如果这些没有这些文件的,终端会通知你,并询问你是否要安装它们,我们只要安装就好了

一旦安装了XCode构建工具和自制程序,你可以重启终端并安装Miniforge:

brew install miniforge

可能有几百MB的下载,所以需要一些时间才能完成。完成之后,为Z shell (zsh)初始化conda:

conda init zsh

最后,重新启动终端。!Miniforge现在已经安装好了,你可以创建虚拟环境了。

第二步-创建虚拟环境

下面的Terminal命令将基于Python 3.8创建一个名为pytorch_env的新虚拟环境:

conda create --name pytorch_env python=3.8

创建完成后,用以下命令激活它:

conda activate pytorch_env

你应该看到这样的东西:

第三步 -安装PyTorch

你可以在Anaconda的网站上找到一个原生的PyTorch包。你应该在平台下寻找osx-arm64,它告诉你它是M1兼容的:

点击该软件包进入详情页后可以看到安装说明,如下所示:

你可以使用下面的命令来安装PyTorch和Torchvision:

conda install -c pytorch pytorch torchvision

PyTorch现在安装好了!让我们接下来测试一下。

第四步 -测试

PyTorch的GitHub页面附带了许多示例,我们找个简单的:用于训练手写数字分类器的脚本https://github.com/pytorch/examples/blob/master/mnist/main.py

只需下载该脚本并从终端运行它:

python main.py

看看输出

看样子还不错!让我们打开Activity Monitor来验证Python是否在本机运行:

如果你在“Kind”下看到“Apple”,这意味着程序是在M1芯片上本地运行的,而不是在Rosetta模拟器下。

最后总结

这样你就可以在M1机器上成功安装PyTorch了。

因为有了miniforge所以安装PyTorch比预期的要容易的多。当然它的运行速度比不上gpu,但这足以让你开始学习和试验。

到此这篇关于M1 mac安装PyTorch的实现步骤的文章就介绍到这了,更多相关M1 mac安装PyTorch内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:

相关文章

  • python发送多人邮件没有展示收件人问题的解决方法

    python发送多人邮件没有展示收件人问题的解决方法

    这篇文章主要为大家详细介绍了python发送多人邮件没有展示收件人问题的解决方法,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-06-06
  • 自己用python做的一款超炫酷音乐播放器

    自己用python做的一款超炫酷音乐播放器

    这篇文章主要介绍了自己用python做的一款超炫酷音乐播放器,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-08-08
  • Python 图片文字识别的实现之PaddleOCR

    Python 图片文字识别的实现之PaddleOCR

    OCR方向的工程师,之前一定听说过PaddleOCR这个项目,其主要推荐的PP-OCR算法更是被国内外企业开发者广泛应用,短短半年时间,累计Star数量已超过15k,频频登上Github Trending和Paperswithcode 日榜月榜第一
    2021-11-11
  • 教你如何使用Python开发一个钉钉群应答机器人

    教你如何使用Python开发一个钉钉群应答机器人

    在聊天工具大肆侵入我们生活各个方面的今天,各种消息无时无刻不在侵扰我们的每一寸时间,这种情况下,一个聊天的机器人就很有必要了.今天,我们来学习一下使用 Python 开发一个钉钉的应答机器人,助你「人生苦短,少回消息」,需要的朋友可以参考下
    2021-06-06
  • Pygame与OpenCV联合播放视频并保证音画同步

    Pygame与OpenCV联合播放视频并保证音画同步

    Pygame的Movie模块已经废弃多年,本文主要介绍了Pygame与OpenCV联合播放视频并保证音画同步,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2021-12-12
  • Python采集腾讯新闻实例

    Python采集腾讯新闻实例

    这篇文章主要介绍了Python采集腾讯新闻实例,一个简单的例子,着重于实现步骤的讲解,需要的朋友可以参考下
    2014-07-07
  • python中获得当前目录和上级目录的实现方法

    python中获得当前目录和上级目录的实现方法

    这篇文章主要介绍了python中获得当前目录和上级目录的实现方法,需要的朋友可以参考下
    2017-10-10
  • Python安装spark的详细过程

    Python安装spark的详细过程

    这篇文章主要介绍了Python安装spark的详细过程,本文通过图文实例代码相结合给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-10-10
  • Python filter()及reduce()函数使用方法解析

    Python filter()及reduce()函数使用方法解析

    这篇文章主要介绍了Python filter()及reduce()函数使用方法解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-09-09
  • Python模块学习 datetime介绍

    Python模块学习 datetime介绍

    Python提供了多个内置模块用于操作日期时间,像calendar,time,datetime。time模块我在之前的文章已经有所介绍,它提供的接口与C标准库time.h基本一致
    2012-08-08

最新评论