MacBook m1安装pytorch
Posted 柳小葱
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了MacBook m1安装pytorch相关的知识,希望对你有一定的参考价值。
🐸今天我们来学习一下安装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的相关链接链接:
- 链接: 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
以上是关于MacBook m1安装pytorch的主要内容,如果未能解决你的问题,请参考以下文章