MacBook m1安装pytorch

Posted 柳小葱

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了MacBook m1安装pytorch相关的知识,希望对你有一定的参考价值。

🐸今天我们来学习一下安装pytorch,博主一般使用的tensorflow,最近和老师探讨一些问题,于是尝试尝试pytorch,今天就在Mac上安装一下pytorch,不知道小伙伴安装之前有没有安装上python,没有的同学可以参考下面这篇文章👇:

🐬PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序。它主要由Facebookd的人工智能小组开发,不仅能够 实现强大的GPU加速,同时还支持动态神经网络,这一点是现在很多主流框架如TensorFlow都不支持的(我保持否定态度!)。 PyTorch提供了两个高级功能: 1.具有强大的GPU加速的张量计算(如Numpy) 2.包含自动求导系统的深度神经网络 除了Facebook之外,Twitter、GMU和Salesforce等机构都采用了PyTorch。

1.pytorch官网

首先给大家几个pytorch的相关链接链接:

2. 创建pytorch虚拟环境

使用以下命令创建pytorch环境

conda create -n pytorch_env

结果如下:

切换到新的环境

conda activate pytorch_env 

结果如下:

到这里,我们的虚拟环境就创建好了,接下来就是安装torch了

3.安装pytorch

使用官网的命令:

conda install pytorch torchvision -c pytorch

结果如下:开始安装一系列的包
我们测试一下:打开jupyter

import torch
x = torch.rand(5, 3)
print(x)

安装成功!

4.anaconda一些命令

conda activate 环境名      激活环境
conda deactivate 环境名。退出环境
conda env list 展示环境
conda install ipykernel 创建内核
python -m ipykernel install --name 名字 内核命名并添加到jupyter

以上是关于MacBook m1安装pytorch的主要内容,如果未能解决你的问题,请参考以下文章

TensorFlow 在训练期间没有使用我的 M1 MacBook GPU

MacBook m1 安装anaconda与pycharm

MacBook m1 安装anaconda与pycharm

M1芯片的MacBook安装docker

M1芯片的MacBook安装docker

如何在 M1 Macbook 上的 Rosetta 2 下运行 Homebrew 安装程序