win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装

win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装

1 Anaconda的安装

在官网下载安装windows版本（64Bit）官网只有最新版Python3.7）

或者在https://repo.continuum.io/archive/下载各种版本的Annaconda，其中：Anaconda3-5.2.0-Windows-x86_64.exe 对应 python3.6.5，选择下载安装。

2 CUDA 、cuDNN的安装

首先查看自己电脑的显卡类型（我的是GTX 960M，安装CUDA9.0没成功，就换成8.0了），
然后，查看显卡版本信息：https://developer.nvidia.com/cuda-gpus

(1)CUDA:https://developer.nvidia.com/cuda-toolkit-archive 选择CUDA Toolkit8.0 - Feb 2017，window-local版本

（2）CuDNN：https://developer.nvidia.com/cudnn（需要注册），如果安装tensorflow-gpu的版本低于1.2.1用得到是cuDNN5.1，更高的tensorflow版本需要cuDNN 6.0。

（3）tensorflow
我没用用国内镜像源，网好的话可以直接下载，我也没有在envs里配置一个tensorflow的环境，直接利用 Anaconda Prompt 或者 cmd 进行安装，这样用起来比较方便：

conda install tensorflow-gpu ==1.3.0

或者

pip install tensorflow-gpu ==1.3.0

直接 conda install tensorflow-gpu 下载的是2.00版本，之前用的cpu版本是这个，好多函数都改了，用起来不是很方便，所以需要指定下载版本。对应关系如下图（不是我的图，原文：https://blog.csdn.net/sunmingyang1987/article/details/102872658）

进行测试：

import tensorflow as tf
sess = tf.Session()
a = tf.constant(10)
b= tf.constant(12)
sess.run(a+b)

以上参考：https://blog.csdn.net/qq_34438969/article/details/88430901

Pytorch 安装

在Anaconda Prompt 中输入：

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

安装速度比较慢，我的 pytorch 下载了两次才成功

安装成功：

测试：

参考：
[1] https://blog.csdn.net/qq_34438969/article/details/88430901
[2] https://blog.csdn.net/sunmingyang1987/article/details/102872658
[3] https://blog.csdn.net/qq_36165459/article/details/88368573