MacBook m1安装pytorch

x33g5p2x  于2021-09-19 转载在 其他  
字(0.9k)|赞(0)|评价(0)|浏览(259)

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

  • 安装python: MacBook m1 安装anaconda与pycharm.

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

1.pytorch官网

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

官网上有一系列安装教程,大家安装gpu版本的可以去看看。

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

相关文章