step 1:
安装anaconda
官网下载有点慢
推荐使用清华镜像+迅雷下载,下载链接如下
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Windows-x86_64.exe
可以在anaconda下创建一个专门用于tensorflow的环境
该环境使用python 3.5.x
anaconda prompt下输入命令:
conda install tensorflow-gpu
安装tensorflow即可
step 2:
安装cuda和cudnn
安装cuda的过程中,选择“自定义安装”,仅勾选cuda即可
(如果选择精简安装,安装过程中会重启,重启过后会无法继续安装)
之后将cudnn文件夹中的三个文件复制到cuda的computing toolkit下
之后,需要设置环境变量,在path中需要包含以下路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
step 3:
在anaconda prompt中,激活之前创建的tensorflow环境
进入python
输入import tensorflow as tf 进行测试
没报错就是安装成功啦!