win10+Anaconda3安装pytorch/torchvision cpu/gpu版本

我在运行一个代码的时候出错提示需要导入torvision，所以就进行了一些百度。因为我之前安装的tensorflow是gpu版本的，然后发现百度上大部分都是cpu版本的安装pytorch方法。最后找到了一种方法可以安装gpu版本的pytorch。首先你需要在nvidia控制面板里查看你的cuda支持哪个版本，然后到https://pytorch.org/进行查找。

因为我的cuda版本是10.1，所以我只能选择10.0的进行下载。如果你是8.0或者9.0的话，只需要将cudatoolkit=10.0中的10.0改成8.0或者9.0。
cpu版本的话基本也差不多，我看了一篇文章说需要把镜像地址改成清华的，所以我们可以在Anaconda Prompt中输入下面几条命令。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
然后根据pytorch官网中的命令安装pytorch。

如果你安装的是cpu版本的，那么你可以在Anaconda根目录下的env中找到添加的框架，如果是gpu版本的可以在pkgs中查找。
这是GPU版本的安装成功的验证提示，如果显示true则表示安装成功。