1. 前言

最近准备开始学习pytorch，所以准备每天学习一个pytorch的函数，并且会将开发过程中遇到的一些困难以及解决办法全部展示出来，方便自己以后查看，也希望和有这方面意向的人多交流和学习。

2. 安装pytorch

  由于一些人（包括我）也是刚入门学习pytorch，所以我们从pytorch的安装开始讲起。

pytorch的安装大家不要去各种神奇的地方看安装攻略，它们写的很多东西都比较老了，直接点击以下官网链接pytorch 官网。向下滑动看到如下所示界面：

然后点击自己想要使用的安装方式，

  • 第一个选项：安装版本，建议安装(1.7.0)，
  • 第二个看自己的电脑是什么就选什么。
  • 第三个注意，conda方式是当你使用anaconda才可以使用conda选项，当然我也推荐使用conda方式，因为anaconda管理python很方便，anaconda安装使用我以前写的博客或者网上找教程就可以。pip是python自带安装方式，如果你不选用anaconda，那么用pip也是可以的。
  • 语言选择python
  • cuda是看你需不需要使用GPU加速，如果你要使用，那么就选择，cuda建议使用10.2，不过如果你同时安装了tensorflow的话，为了使你的tensorflow还可以继续使用GPU加速，建议使用10.1（因为TensorFlow目前最多支持到10.1）。然后如果你从来没哟使用过GPU加速，光安装cuda是不够的，我们还需要安装cudnn，这个东西是让你的显卡具备科学计算的能力，安装教程可以百度搜索，翻我以前的也可以，就是比较老了，还是建议百度。
  • 然后在你的anaconda prompt里面输入run the command里面的命令即可。
    安装过程比较慢，请耐心等待。

安装好以后，本次大部分使用jupyter notebook来进行演示，当然后面可能会用vscode（因为本人转向了vscode，而且真的好用，有时间我出教程如何使用vscode来编写，当然pycharm的优点是便于调试；缺点是启动慢，占内存）

3. 使用jupyter notebook

  • 电脑点击启动按键，输入jupyter notebook即可

    选择第一个，点击后看到如下界面：

    点击new，选择python3看到如下界面：

    然后开始使用，输入代码，首先测试pytorch是否安装成功：
    安装cpu输入
import torch

按(shift + enter)运行，没报错即为正常。
安装GPU版本输入：

import torch
print(torch.cuda.is_available())

输出为True即为正常，输出为False说明cuda安装失败。如果报no module named "torch"则是因为pytorch没有安装成功，如果显示缺少了一个dll文件，那么是因为C++库没有安装，点击以下官方链接visual C++下载，进去后下载x86与x64文件，依次安装x86与x64文件，然后重启电脑即可解决：

至此pytorch安装成功，然后开始pytorch学习之旅。