PyTorch安装教程（带图文）

---

前言

上一篇文章讲解了如何安装Anaconda,现在来安装pytorch,安装好工具就可以好好学习了。

---

1.准备工作

由于我安装anaconda时选择了最新版的，它默认安装了最新版python 3.8.5版本，我看网上大多数博主推荐python 3.6，说兼容性较好。所以需要再安装一个python 3.6。单击启动Anaconda prompt,如下图

运行

conda create -n py36 python=3.6

这是创建环境的命令，-n是指name, py36是指环境指定命名为py36,python=3.6是指安装的是python 3.6。
遇到下图

输入

y

Python3.6环境安装完成，接下来激活环境，运行

Conda activate py36

环境由默认的base转为了py36，如下图

准备工作已完成，Anaconda prompt窗口不要关闭，最小化即可，接下来就是安装pytorch.

2.安装pytorch

下载地址：pytorch官网
进入官网，点击Get Started,鼠标往下滑，根据自己的计算机环境选择，我的电脑是windows版，选择stable,windows,conda,python,电脑若有GPU，可选择CUDA10.2或是CUDA11.1。电脑无GPU，则选择CPU.我电脑带有GPU，我选择CUDA10.2。如下图

GPU是显卡上的核心单元，是一种专门做图像和图形相关运算工作的微处理器。如何判断自己电脑是否有GPU？第一种方法：打开任务管理器，找到性能，如下图，我电脑带有两个GPU,一块集成显卡，一块独立显卡。

第二种方法：打开cmd,运行

nvidia-smi

如果Driver version的数值大于390即可选择电脑带GPU的选项，如下图

选择好后，复制run this command右边NOTE下面部分的命令到之前打开的Anaconda prompt中运行，如下图


遇到下图

输入

y

安装过程有点久，需要耐心等待。安装完成会出现如下语句

如果安装过程中网速不好，或者其他原因，导致部分包安装中断，可以运行上一条复制的那条命令。
由于我安装过程中笔记本没有电自动关机，导致部分包没有安装完成。如下图
所以我重新运行了那条命令。

3.检验

安装完成后开始检验，在Anaconda promp中的py36环境下运行：

python

回车后继续运行

import torch

回车后，如果没有报错，则说明pytorch安装成功。
查看是否可以利用GPU，运行

torch.cuda.is_available()

显示Ture则说明pytorch可以使用GPU。如下图

pytorch安装成功。

4.总结

以上就是pytorch的整个安装教程。安装重点在如何在Anaconda中安装其他python版本，以及根据自己电脑配置选择安装pytorch的版本。其他的应该不是什么大问题。
希望本篇博文对你有所帮助，欢迎小伙伴们关注、点赞、收藏、评论，这对我一个新人博主很重要，将鼓励我继续下去。

参考资料
1.PyTorch安装最全流程
2.Anaconda以及PyTorch安装总结
3.PyTorch深度学习快速入门教程