Pytorch教程(一)—— 安装篇

写在前面

最近帮老师给实验室的师兄弟姐妹们(怎么叫都不太对的样子233)整理一下pytorch的教程 ,顺便自己也巩固下这方面的知识。这个教程主要会分为几个部分:

  1. 安装篇
  2. 搭建网络篇
  3. 数据处理篇
  4. 训练篇
  5. 其他

除了像其它普通博客一样介绍一些基本的知识以外,我还会结合自己比赛、科研的一些代码经验进行一些拓展。该系列教程不光适合初学者,对于有一些知识储备的同学来说我相信也会有一定意义的。当大家看完这部分之后也可以参考我的往期博客做一些小实战。

  • GAN生成MNIST数据集(pytorch版)
  • kaggle入门比赛Dogs vs Cats(pytorch实现)

1. 进入官网

Pytorch官网:https://pytorch.org/

2. 按照选项选择

首页往下滑动,或者点击get start,就可以看到下面的选项:
Pytorch教程(一)—— 安装篇_第1张图片
根据自己的情况进行选择,我这里的选择是系统为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命令。
Pytorch教程(一)—— 安装篇_第2张图片

你可能感兴趣的:(深度学习)