win10+anaconda3安装tensorflow-gpu和pytorch-gpu版

1、需安装的东西

• Win10 x64位
• Anaconda3 x64 python 3.6.5
• CUDA Toolkt10.0
• cuDNN 7.5 for cuda10
• Nvidia drive 430.64
• Visual Studio 2017 community

2、具体实现

2.1 windows10的安装

在我之前写过的安双系统的一篇文章里面有提及安装win10的教程在此不作过多赘述

2.2 Anaconda3的安装（略）

安装过程中提示“不建议添加到环境变量中”，那个默认不勾选就好，要启动到开始里面，到Anaconda Prompt里面用管理员身份打开即可

2.3 Visual Studio 2017 community

现在官网上面已经提供最新的2019下载，之前的版本下载好像要订阅什么的，你们也可以下载我这里提供的链接：https://pan.baidu.com/s/1Zvtg8nHPC_oZBA9rRC_CAA
提取码：s90s ，安装没什么难的，主要之后的显卡驱动要用到VS里面的东西所以要安装它。

2.4 Nvidia drive 430.64的安装

其实我感觉这个是不必须的，因为youtube上面很多教程都没有提及这个，而且我发现后来安装CUDA Toolkit的时候它也会装drive，但是我是装了，你们可以不装试试看
去这里找你电脑的N卡对应的驱动，然后下载安装，这个过程可能需要耐心等待，其中选项都默认就行

Nvidia驱动正在安装

下面是我安好的截图

显卡驱动安装成功

安好之后桌面上多了这样一个图标

驱动图标

之后需要重启（在我不重启的时候后面的CUDA10.0安装不成功），至此显卡驱动已安装完成！

 

2.5 CUDA Toolkt10.0的安装

去这里下载对应版本的，我是win10本地安装，如果不想安装10.1可以在Release Notes里面找10.0版本的安装

CUDA Toolkit10.0

这个是安装中的截图：

cuda10.0安装过程图

 

重启后cuda10安装成功

 

这里面都安装在C盘，默认没有改安装路径，安装成功桌面上出现这样的图标：

cuda10.0安装完图标

2.6 cuDNN 7.5 for cuda10的安装

去这里注册下载，这个下载得先注册，选择cuDNN 7.5 for cuda10，因为我们之前装的就是CUDA10.0

cuDNN7.5.1下载

下载完就是一个压缩包

cudnn.png

，解压出来就是一个名为“cuda”的文件夹，里面有

解压出来的cuda中的文件

 

• 有些教程里面是说把这三个文件夹里面的.dll文件覆盖掉cuda安装目录下的对应文件，我分析他们之后就没有添加cuda的环境变量了，我没试过
• 我是按照把C:\cuda\bin添加到环境变量里面了，我认为一样的，我实践证明有效

至此要安装的东西几乎完成了

3、 添加环境变量

除了上面说的把C:\cuda\bin添加到环境变量里面，还要照下图的添加两个环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp

这就是刚刚CUDA的安装目录（当然上面是我自己的啦，你们要看自己的改一下）

4、安装packages

4.1 安装tensorflow-gpu版

进入到Anaconda Prompt里面（因为我没有安装python，这个是anaconda自带的），因为anaconda包挺齐全的，所以直接安装

pip3 install tensorflow-gpu

我是下载下来.whl文件，然后cd到.whl目录，用

pip install tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl

这中间我出现过ModuleNotFoundError: No module named 'numpy.core._multiarray_umath' 错误，这个只要卸载numpy再重新安装一下就好了

pip uninstall numpy
pip install numpy

下面我测试了一下，看看是否安装成功了

 

测试tf安装成功

4.2 安装pytorch-gpu版

一般都是到官网去用一句话安装，如

pytorch官网

但是这上面一般都是最新版的，点击下面的Previous versions of PyTorch，可以找到对应平台的.whl文件下载下来，和上面安装tensorflow的.whl文件一样安装pytorch。

pytorch_gpu安装成功

转自简书--赖子NEU