win10下搭配tensorflow环境----心碎版
本机配置:cpu i3-7100 ,GTX650
安装所需软件:visual studio--2015版,cuda_9.0.176_win10.exe ,cudnn-9.0-windows10-x64-v7.zip ,tensorflow_gpu-1.5.0-cp35-cp35m-win_amd64.whl (pip包阿里云镜像里面下载)
软件依照顺序安装(注意在安装完cudnn后,要用vs2015编译一下一个文件,然后再用pip install tensorflow-gpu)
- 利用VS2015编译测试文件
打开C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0,找到对应VS版本的sample,本例中为Samples_vs2015.sln,双击打开:
选择Release,X64
右键1_Utilities,点击生成(U)
成功编译出现图片下方文字:成功5个…
至此,“C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64\Release”文件夹中会出现我们需要的deviceQuery和bandwidthTest。
(3)验证deviceQuery和bandwidthTest
打开命令提示符:定位到 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64\Release目录,分别输入deviceQuery,bandwidthTest并运行,出现如下类似信息便说明CUDA安装成功。
(4)环境变量
- 在用户变量,Path中添加:C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0\bin
- 确认系统变量中:CUDA_PATH和CUDA_PATH_V9.0已经存在
最后------测试一下在python下能否import tensorflow就ok啦