安装Anaconda

image-20240909000959700

安装Pytorch

使用Anaconda新建Pytorch环境,激活新环境以后安装Pytorch

image-20240909001137964

这里不要使用conda安装,否则后续的torch.cuda.is_avilable()**的返回值都是FALSE**,会出现下面的问题。需要在安装好的conda中使用pip安装,CUDA版本可以向下兼容。

image-20240909002224831

PyTorch与NumPy函数区别

大部分的PyTorch与NumPy函数是相同的,矩阵全部变为张量,部分函数的名字不同,但功能基本一致。

np 变为 torch、array 变为 tensor

image-20240924140829047

PLT使用

下面的网站有很多PLT的案例,可以替换成自己的数据以后直接使用 Examples — Matplotlib 3.9.2 documentation

使用PLT有两种方式,一种是MATLAB方式,一种是面相对象的方式。MATLAB的方式更全面,也更符合MATLAB使用的习惯。

image-20240924142949129

各层大小计算

卷积层:

image-20240925183018383

池化层:

image-20240925182956352

案例:

image-20240925183058663