Win10 TensorFlow--gpu

来源:互联网 发布:西部数码域名 编辑:程序博客网 时间:2024/06/06 02:28

安装前准备

  • 下载VS2015,建议安装社区版,免费,安装时选择自定义,然后只安装visual studio C++,别的都不装,千万别装成全家桶
  • 下载DXSDK_Jun10.exe,安装时直接next就行,下载地址http://pan.baidu.com/s/1kUU4CCZ
  • 下载CUDA,cudnn,地址http://pan.baidu.com/s/1i5A6e3b
  • python3.5(Anaconda4.2)

安装CUDA及cudnn

  • 安装CUDA,直接下一步
  • 解压cuDNN到一个你喜欢的位置,复制文件夹的绝对路径并加到 PATH 环境变量去,然后将刚才解压出来的下面这些文件复制到相应位置,我解压在了E盘根目录下了。
解压后的文件路径 复制-> 目标文件夹路径 E:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin E:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include E:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

编译cudnn示例程序

(1)在 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0 目录下,有CUDA的示例程序。由于我安装的是VS2015,所以我打开Samples_vs2015.sln那个解决方案文件,将解决方案配置更改为Release和x64.
使用Release模型,由于所有程序安装包用的都是64位版本,所以改为x64比较好。

(2)右键单击,编译整个解决方案。如果不出意外,将会编译成功。如果提示是缺少:
“d3dx9.h”、”d3dx10.h”、”d3dx11.h”头文件 ,说明前面安装DirectX SDK没有安装好,重新安装下DXSDK_Jun10.exe,再次编译。

(3)关闭VS2015,在
c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release
目录下找到deviceQuery.exe这个文件。打开一个cmd窗口,定位到 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release目录,输入:deviceQuery.exe ,然后回车。会得到如下结果:
安装成功图片

如果你的结果与上面类似,恭喜你!CUDA 8.0安装成功了!

安装tensorflow1.1.0

激活python3.5环境,然后cmd输入
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl

接下来随便写个程序检验下就完成啦

原创粉丝点击