目录
[1.1 torch介绍](#1.1 torch介绍)
[1.2 torch.nn相关库的导入](#1.2 torch.nn相关库的导入)
一、pip安装torch
1.1 torch介绍
torch的基本功能:
**①torch:**张量的相关运算,例如:创建、索引、切片、连续、转置、加减乘除等相关运算。
**②torch.nn:**包含搭建网络层的模块(modules)和一系列的loss函数。
例如:全连接、卷积、池化、 BN分批处理、dropout、CrossEntropyLoss、MSLoss等。
包含五大基本功能模块:nn.Parameter、nn.Linear、nn.functional、nn.Module、nn.Sequential
**③torch.autograd:**提供Tensor所有操作的自动求导方法。
**④torch.nn.functional:**常用的激活函数relu、leaky_relu、sigmoid等。
**⑤torch.optim:**各种参数优化方法,例如:SGD、AdaGrad、RMSProp、Adam等。
**⑥torch.nn.init:**可以用它更改nn.Module的默认参数初始化方式。
**⑦torch.utils.data:**用于加载数据。
1.2 torch.nn相关库的导入
torch.nn库是专门为神经网络设计的模块化接口,nn构建于autograd之上,nn是Neural Network的简称,可以用来定义和运行神经网络,帮助程序员执行下述与神经网络相关的行为。
(1)创建神经网络
(2)训练神经网络
(3)保存神经网络
(4)恢复神经网络
其中torch.nn相关库的导入如下所示:
python
import numpy as np # numpy数组库
import math # 数学运算库
import matplotlib.pyplot as plt # 画图库
import torch # torch基础库
import torch.nn as nn # torch神经网络库
import torch.nn.functional as F
1.3win10上torch的安装命令
python
pip install torch===1.7.1 torchvision===0.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
出现如下界面即可:
注意:[Pytorch系列-30]:神经网络基础 - torch.nn库五大基本功能:nn.Parameter、nn.Linear、nn.functioinal、nn.Module、nn.Sequentia
二、pip安装Opencv
(1)win+R输入cmd;
(2)然后切换到anaconda安装的位置,然后进入Script文件夹的位置。
(3)依次输入下述指令:
python
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install opencv-contrib-python -i https://pypi.tuna.tsinghua.edu.cn/simple
(4)输入conda list检查是否下载成功,出现下列标识即可。
三、pip安装scipy库
(1)安装Numpy:因为Scipy依赖于Numpy,所以需要先安装Numpy,命令如下:
python
pip install numpy
(2)安装Scipy
python
pip install scipy