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下

win10安装tensorflow gpu

之后,需要设置环境变量,在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 进行测试

win10安装tensorflow gpu

没报错就是安装成功啦!



相关文章: