windows系统下用conda安装gpu版本的pytorch

在自己经历了装的乱七八糟导致重装系统之后。我痛定思痛，必须要在win10上把pytorch装上。以下是我自己走了弯路成功之后的总结，希望可以帮到大家一次成功。

   

过程就是三步走：

      ①安装cuda

   这里挺有“学问"的，如果你曾经自己装过cuda，你会发现在cmd运行nvcc -V 和运行nvidia-smi给出的cuda版本是不一样的。比如我自己在第一次尝试装pytorch的时候，我下载的cuda9.2。用nvcc -V给出的结果是9.2版本，用nvidia-smi的话，给出的结果是10.2版本，当时也很困扰我，直接导致我重装系统。今天我看到一篇帖子，说的是这两个命令给出的结果，一个是运行时环境的版本号，一个是驱动的版本号。总是就是一句话，以你在控制面版看到的cuda版本，也就是nvcc -V的cuda版本，也就是你自己下载的cuda为准。

    那么去哪里下载呢?答案是英伟达官网。直接百度cuda10.1进入官网下载即可。P.S.：如果你百度的是”cuda“，那么会进入英伟达最新的cuda11.1的网址，我不知道怎么找到10.1的，索性在地址栏补充完整了。

    我选择的版本如下图所示：

    在你下载安装好cuda10.1，并且验证安装成功（验证方法上文说了）之后，第一步完成

②在conda中建立虚拟环境

    这里假设大家都安装了conda。建立虚拟环境是为了避免装的乱七八糟把base环境搞崩了。

conda create -n xxx python=3.5

这里的xxx是你自己定义新的虚拟环境的名字，后边的python版本你可以自己修改成想要的，这里我用的是3.5版本。安装成功后，运行

conda info -e

就可以看到除了一个base之外，还有你新创建的虚拟环境。接着运行conda activate gputorch（虚拟环境名），进入虚拟环境

③安装pytorch

    百度搜索pytorch进入官网，因为我第一步安装了cuda10.1，所以这里我要对应选择10.1

然后，最下边会给出一个conda安装的命令。我们把它复制到conda窗口。注意：此时需要把

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

中的 -c pytorch去掉。因为，如果过不去的话，是用官方的源下载，速度会很慢。如果删除的话，就会用你之前在conda中配置好的国内的源下载。（比如清华的源，如果不会配置源的话自己百度一下，这里不再赘述）

    不出意外下载会很快然后会提示安装成功。这个时候我们测试一下就好。在当前的虚拟环境中打开python，运行import torch验证pytorch是否安装成功，再运行torch.cuda.is_available()测试是否可以调用gpu。

    如上图所示，返回true则安装成功。