M1mac安装PyTorch的实现步骤
⽬录
第⼀步 -安装和配置Miniforge
第⼆步-创建虚拟环境
第三步 -安装PyTorch
第四步 -测试
最后总结
M1 macbook已经不是什么新产品了。TensorFlow官⽅已经给出了安装指南和效率评测。
本⽂将介绍如何在M1机器上本地安装和运⾏PyTorch。你使⽤的M1机型(Air、Pro、Mini或iMac)没有区别。
第⼀步 -安装和配置Miniforge
我花了很多时间为数据科学需求配置我的M1 Mac。但是都不能完美的解决我的问题。直到我到了这个。根据⽹速的不同,完全设置需要5到10分钟。
⾸先,你需要安装Homebrew。它是Mac上的⼀个包管理器,你可以在终端上执⾏以下命令来安装它:
/bin/bash -c "$(curl -fsSL raw.githubusercontent/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现在安装好了!让我们接下来测试⼀下。
python安装教程非常详细
第四步 -测试
只需下载该脚本并从终端运⾏它:
python main.py
看看输出
看样⼦还不错!让我们打开Activity Monitor来验证Python是否在本机运⾏:
如果你在“Kind”下看到“Apple”,这意味着程序是在M1芯⽚上本地运⾏的,⽽不是在Rosetta模拟器下。
最后总结
这样你就可以在M1机器上成功安装PyTorch了。
因为有了miniforge所以安装PyTorch⽐预期的要容易的多。当然它的运⾏速度⽐不上gpu,但这⾜以让你开始学习和试验。
到此这篇关于M1 mac安装PyTorch的实现步骤的⽂章就介绍到这了,更多相关M1 mac安装PyTorch内容请搜索以前的⽂章或继续浏览下⾯的相关⽂章希望⼤家以后多多⽀持!

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。