安装pytorch gpu版本

在windows下用anaconda虚拟环境安装pytorch gpu版 成功！

接下来介绍安装过程。

推荐全部看完本文再进行安装
可以重点看黑体字，0部分介绍pip安装，后面的部分介绍conda安装，pip部分简明扼要，conda部分介绍的更详细些。祝你安装成功。

0. 用pip安装

pip比conda 速度更快一些，更随心所欲一些，所以介绍一下pip安装，直接上代码

conda create -n pytorch18 python=3.8 # 创建pytorch的虚拟环境
activate pytorch18  # 激活pytorch环境

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
# 上面的安装代码去https://pytorch.org/get-started/previous-versions/找，根据自己想用的pytorch版本和cuda版本选择相应的命令。关于如何查看cuda版本可以在我目录中的第三部分去查看。
# 安装过程会发现安装很慢，这时候ctl+c取消安装，找到这个下载速度很慢的安装包的来源网址，这个网址在安装界面中会显示，复制，去浏览器中打开，然后crl+f，输入安装包的名字，可以找到需要的安装包，然后直接离线下载到本地。

pip install E:\pytorch\torch-1.8.0+cu111-cp38-cp38-win_amd64.whl  # 后面的路径就是安装包在的路径，直接安装

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
# 再次输入这个命令来安装另外两个模块。搞定

pip install ipython  # 安装ipython测试是否安装成功
ipython # 打开ipython
import torch
torch.ones(8) # 创建一个tensor
torch.cuda.is_available() # 测试gpu是否可以使用，成功的话会返回True

1. 安装准备

1.是windows系统下的安装

2.其次是在anaconda的虚拟环境中安装，anaconda3的安装以及如何创建虚拟环境就不讲了。
使用conda前在base环境下进行大扫除：
清除下缓存信息

conda clean --packages --tarballs

2. 安装指令

3.pytorch官网（官网链接）讲了如何用conda安装。如下图，，如果你想安装其他版本，也可以点击图片中右上角的install previous versions of pytorch

输入蓝框中的pytorch安装指令进行安装。
需要选择cuda版本，那么如何查看cuda的版本呢？

3. 查看、提升cuda版本

4.nvidia显卡是否支持cuda？如何查看cuda版本？如何提升cuda版本？

3.1 判断显卡是否支持cuda

下图显示了支持cuda的显卡

3.2 查看 cuda版本

桌面右击打开nvidia控制面板，点击左下角系统信息，点击组件，在3D设置里的第三个就是cuda版本了，我的是8.0.0。

3.3 提高cuda版本

驱动版本和cuda版本是有关系的，因此下载更高版本的驱动程序就可以得到更高版本的cuda。下图显示了驱动和cuda版本之间的关系。

3.4 查看驱动版本

如下图，在nvidia控制面板的系统信息中，可以查看驱动程序版本。
nvidia官网没有旧的驱动程序下载，可以到这个网站https://down.gamersky.com/oth/soft/下载。

4. 离线下载torch安装包

现在可以输入pytorch安装指令进行安装了。但是下载很慢。为了避免连接中断导致下载失败，所以离线下载torch安装包，分两步操作：

4.1 离线下载安装包

输入pytorch安装指令后，会显示各个包的下载地址（如下图中蓝框）地址显示的pytroch指的就是https://conda.anaconda.org/pytorch/win-64/这个网站。直接打开这个网址下载pytorch安装包。迅雷两分钟搞定。

与此同时我们需要输入ctrl+c来关闭刚刚进行的安装，我们只是为了看下pytorch安装包的下载网站。接着要输入下列指令，来清除刚刚下载所产生的缓存文件。

conda clean --packages --tarballs

安装包下载下来之后，将pytorch安装包复制到 \Anaconda3\pkgs 中去。

4.2 修改urls.txt文件

第二步：上图中最下面有一个urls.txt文件，将迅雷中pytorch安装包的下载链接复制到urls.txt文件中。

从下图可以看到我复制在了第一行。这样pytorch安装包就不需要下载了

5. 安装及检查

输入pytorch安装指令，回车。

检查是否安装成功，同样pytorch官网也给出了方法如下图所示：

恭喜你，pytorch安装成功！有问题私信或留言。