MacBook M1 PyTorch 安装方法

Mac的pytorch安装较Windows要简单一些,我这里使用的是MacBook Air M1,其无法使用英伟达GPU,所以下载的pytorch不能使用cuda,同时我已经提前下载好了pycharm,需要的伙伴请自行下载~。

1. 下载anaconda

anaconda包含了许多非常有用的包,如之后需要用到的conda,所以首先我们需要下载anaconda,直接上官网下就行,链接如下:

Anaconda | The World’s Most Popular Data Science Platform

下载好之后按照提示一步步进行安装。

2. 安装pytorch

(1)首先查看当前python版本,并创建一个torch的虚拟环境,命令如下:

conda create -n 你想起的环境名字 python==当前版本

(2)进入该虚拟环境,命令如下:

conda activate 环境名字

该命令运行之后,在最左边括号中将显示创建的环境名称。

(3)下载pytorch

前往torch官网,找到合适的下载命令,链接如下:

Start Locally | PyTorch

 

 

在创建的环境中输入命令开始下载,官网下载速度较慢,我花了大约一个小时下载完成。 

 

(4)进行测试

输入python3命令进入python,依次输入以下命令测试pytorch是否成功下载安装:

import torch
torch.device('mps')
torch.__version__

 若下载成功,将会显示出对应的torch版本。

3. 在python项目中应用torch环境

在pycharm新建的python项目中使用我们已经建立好的torch环境。

 

 

在python console中进行测试。

 

 

测试成功,由于Mac没有英伟达GPU所以cuda能否使用方法返回值为false。 

  • 3
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值