写在前面
最近帮老师给实验室的师兄弟姐妹们(怎么叫都不太对的样子233)整理一下pytorch的教程 ,顺便自己也巩固下这方面的知识。这个教程主要会分为几个部分:
- 安装篇
- 搭建网络篇
- 数据处理篇
- 训练篇
- 其他
除了像其它普通博客一样介绍一些基本的知识以外,我还会结合自己比赛、科研的一些代码经验进行一些拓展。该系列教程不光适合初学者,对于有一些知识储备的同学来说我相信也会有一定意义的。当大家看完这部分之后也可以参考我的往期博客做一些小实战。
1. 进入官网
Pytorch官网:https://pytorch.org/
2. 按照选项选择
首页往下滑动,或者点击get start,就可以看到下面的选项:
根据自己的情况进行选择,我这里的选择是系统为Windows,计算平台为CUDA10.2(根据自己的CUDA版本决定,GPU没有cuda就安装CPU版本)
3.安装
复制“Run this Command”中的命令到cmd命令窗口中进行安装。比如我这里的命令就是conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
4.验证
打开cmd命令窗口,输入python
命令,进入python编程模式。
输入import torch
后敲回车,如果没有报错说明安装成功,另外还可以输入torch.__version__
查看pytorch命令。