Windows10安装pytorch和tensorflow

参考： https://blog.csdn.net/qq_39557388/article/details/104477244?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase

参考自一个说话很可爱的小姐姐。

事前了解1：python包管理工具：Conda和pip

https://www.jianshu.com/p/5601dab5c9e5

Conda 是一个开源的软件包管理系统和环境管理系统，用于安装多个版本的软件包及其依赖关系，并在它们之间轻松切换。

事前了解2：安装anaconda，cuda，cudnn

在安装pytorch前，
安装anaconda，cuda，cudnn。

一.如何检查电脑是否有合适的GPU？

（1）如果能找到NVIDA控制面板，则说明该电脑有GPU。

第一种方法：打开在桌面上单击鼠标右键，打开NVIDIA控制面板。

如果你单击鼠标右键你并没有发现这个玩意儿，请先去下载“驱动人生”，安装好打开软件，然后等它检查完毕，它会告诉您您的计算机显卡没有驱动（或者告诉您“您的计算机硬件驱动异常”之类的），直接无脑点击”一键修复“即可。
然后，这时所有的亲们单击右键都能找到NVIDIA控制面板咯。

第二种方法：

打开NVIDIA控制面板后：

（2）点击<帮助>然后点击 <系统信息> 弹出下面的对话框，在驱动程序版本那一栏就能看到该计算机使用的驱动版本。

我这里是：417.35，根据实际情况而定。

（3）打开它，点击左下角的“系统信息”，然后点击”组件“，在”3D设置“里有个"NVCUDA.DLL"，请您向右看，是不是有”NVIDIA CUDA 数字 driver"？

这个数字就是跟你的gpu匹配的CUDA版本号，您只能用这个版本的CUDA或者cudatoolkit来匹配您的gpu。

我这里是10.0.132，根据实际情况而定。

二.安装Pytorch

（1）在线安装过程

准备

pytorch官网：https://pytorch.org/

打开https://pytorch.org/，第一行选stable，第二行选windows，第三行选conda（建议选conda，pip也行），第四行选python

重点来了，如果第五行有你的显卡对应的版本最好，直接选上，然后复制安装命令。

如果没有，请点击上面那段话中的橘黄色字install previous vision of pytorch，找到适合您显卡的cuda版本，复制安装命令。

我查到的是

conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=10.0 -c
pytorch

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

第一步，打开Anaconda Prompt，创建名为pytorch的环境。

conda create --name pytorch python=3.7

conda 是指调用 conda 包，create 是创建的意思，–name 是指后面的名字是屋子的名字， pytorch是屋子的名字（可以更改成自己喜欢的），python=3.7 是指创建的屋子，是 python3.7 版本。

第二步，进入这个毛坯房。输入：

conda activate pytorch

第三步：从云端请pytorch到家里来之前添加几个镜像用于加速，建议一个一个加，防止意外。。。。（也可以忽略这个步骤，用原来的源，速度慢而已）

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

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

第四步，执行之前查到的代码：

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

注意：将命令最后面的 -c Pytorch 去掉，这样才会在我们给出的下载源去下载相关软件包，如果不是，则不用去掉

（2）离线安装过程（推荐！！！！本人用此方法下载成功）

如何离线安装PyTorch

第一步，下载PyTorch安装包

https://download.pytorch.org/whl/torch_stable.html

还是慢。

第二步，安装

分别输入pytorch和torchvision的安装命令：
安装pytorch： pip install (D:)（torch-1.3.0+cpu-cp36-cp36m-win_amd64.whl）
安装torchvision： pip install (D:)（torchvision torch-1.3.0+cpu-cp36-cp36m-win_amd64.whl）

（3）验证是否安装成功

进入python，输入：

python

然后输入：

import torch

如果没报错，就安装成功了。

然后检查pytorch是否能用GPU加速计算，继续输入：

print (torch.cuda.is_available())

返回true就是可以。

四.如何卸载pytorch?

按顺序执行：
1、使用conda卸载Pytorch

conda uninstall pytorch
conda uninstall libtorch

2、使用pip卸载Pytorch

pip uninstall torch

3、下载失败

conda clean -p