创建conda环境

conda create -n pytorch
conda activate pytorch

注意下面的安装的都是linux的cpu版本的pytorch,更多版本和命令参考官网

使用conda安装pytorch

conda install pytorch torchvision torchaudio cpuonly -c pytorch

或者使用pip安装pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

把环境添加到jupyter的环境里

conda install ipykernel
python -m ipykernel install --user --name pytorch --display-name "pytorch"

检测是否安装成功

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

输出数字的矩阵，表示安装成功，报错说明安装没有成功。

使用conda安装报错，找不到pytorch channel,使用下面的命令添加中科大的conda源，再安装即可。

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision torchaudio cpuonly pytorch

pytorch入门

1. 张量tensor

1个数就是标量，0维张量
1列数就是矢量，1维张量
1个数据表，矩阵，2维张量
1堆数据表，矩阵数组，3维张量

低纬度的张量

用正方体表示3维张量

image.png
一列正方体，就是4维张量
一片正方体，就是5维张量
image.png

2. 数据集和数据加载器

3.变换

4.构建模型

5.自动求导Autograd

6.优化模型

7.保存和加载模型