3.4 深度学习库PyTorch安装「梗直哥深度学习公开课」
PyTorch就是一个计算工具,你能够借助它,使计算机完成复杂的计算。
Py就是Python,Torch字面意思是火炬,我们可以把它理解成能在GPU中计算的矩阵,而且运算速度特别快。
为了能在后续的课程中,更好的学习Pytorch,本小节我们来学习如何安装它。
3.4.1 下载与安装
首先访问pytorch官网（pytorch.org）,点击install
先选择pytorch的版本,这里我们选择最新的稳定版1.11; 接着是操作系统,我这里使用的是windows;安装方式 我使用conda进行安装; 编程语言选择python;
计算平台 我选择新版的Cuda11.3
这里如果你对conda 和 cuda还不了解,建议你先看一下这一章的前两节教程。
我来捋一下 pytorch cuda 与 Nvidia之间的关系：我们想安装PyTorch ，使用GPU进行快速运算，那么就需要先安装CUDA，因为它是用来管理GPU实现并行运算的。而CUDA toolkit是Nvadia公司的推出的，所以安装cuda之前，需要确保你的电脑有英伟达独立显卡。
同时要注意下载的版本问题，要根据显卡驱动的版本，下载支持的Cuda 版本。如果有必要，可以到这个地址（https://www.nvidia.com/Download/index.aspx?lang=cn） 先升级你的显卡驱动，再根据Cuda 版本安装支持的Pytorch版本。
如果你的电脑没有英伟达独立显卡，那么安装CPU版本也是可以的。在学习阶段，使用CPU也能支持跑个小实验。但是你要知道实际项目中还是使用GPU来训练模型。
然后复制下面这条生成好的命令
打开anaconda prompt，使用conda创建一个新的虚拟环境，用来接下来pytorch的学习：
conda create -n pytorch python=3.8
-n 后面是虚拟环境的名字 ，我这里叫pytorch ，当然你也可以叫别的，python的版本指定了3.8
环境创建好后，使用：conda activate pytorch 激活这个环境，可以看到此时括号里就是pytorch
接下来，把从网站上复制的安装命令，粘贴在这，回车。剩下的就是等待整个安装过程。它会安装pytorch以及其关联的包。
如果要安装之前版本的Pytorch，请访问INSTALLING PREVIOUS VERSIONS OF PYTORCH,找到要安装的版本，进行安装。
3.4.2 检测是否安装成功
pytorch安装完成后，需要验证是否安装成功。
先在这个环境中启动python：输入python，回车
然后导入torch模块：import torch 回车. 注意是torch而不是pytorch
如果像我这样，界面没有报任何的错误信息，那么意味着PyTorch已经顺利安装了
我们来创建一个torch张量，输入：x = torch.rand(5) ，打印x，也可以正常工作
现在我们来检查CUDA，也就是GPU是否可用
输入：torch.cuda.is_available() 回车
如果输出的是True，意味着可以使用GPU，大功告成
如果你安装的明明是 GPU 版本，但是返回的却是“False”，GPU 不可用。那么你就要对照着上面讲解pytroch cuda 与 nvidia关系的地方检查，一定可以找到原因的。
关于PyTorch的视频安装讲解请看：Bilibili:梗直哥丶
Github:梗直哥
机器学习必修课：经典算法与Python实战
