windows10 安装tensorflow(GPU版本)

windows10 安装tensorflow(GPU版本)

tensorflow安装

一 安装cuda，cudnn

1. 下载cuda8.0，找对应版本。文件是.exe格式，直接双击安装，按提示下一步装到底就行。装完测试一下是否安装成功，命令行输入 nvcc -V ，看到版本信息就表示安装成功了。
2. 下载cudnn6.0，下载相应版本。tensorflow版本不同，需要的cuDNN版本也不一样，具体看tensorflow的发布说明，如:tensorflow1.3的发布说明

二 配置cudnn

下载到对应版本的cuDNN，解压：
1. 将解压后的bin目录配置到系统环境变量(path)中
2. 将相应的文件拷贝到相应的目录：
copy cudnn\bin\cudnn64_5.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
copy cudnn\include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
copy cudnn\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

三 安装tensorflow

1. pip install –upgrade tensorflow-gpu(安装最新版本tensorflow)
2. pip install tensorflow==1.3(安装指定版本tensorflow)