在网上搜了一大堆教程,教怎么安装tensflow, 但是我总是出错,将自己的坑记录下来;
安装环境: win 10
python: 3.5
第一步:安装python3.5 (不要安装3.6)
安装pip
以上两步记得将环境变量加载进去:
C:\Users\Administrator\AppData\Local\Programs\Python\Python35
注意pip的环境变量加载如下:
C:\Users\Administrator\AppData\Local\Programs\Python\Python35\Scripts\
第二部安装cuda和CUDNN
记得将cuda加入环境变量中去:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
下载CUDNN.然后解压,分别将 bin lib include 问价夹下面的文件拷贝到cuda中去,也就是拷贝到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\下面的bin lib include 中去。
第三步:安装VS2015(VS2010, vs2012,vs2013,vs2015应该都可以),编译cuda测试模块,查看cuda 是否安装成功,方法如下:
编译完成后,Win+R打开命令行窗口,cd C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release,运行deviceQuery,如果显示如下画面,则安装成功。
第四,pip 安装tensorflow
使用管理员模式运行cmd(可以在开始菜单里右键管理员模式打开)
输入代码
pip install --upgrade --ignore-installed tensorflow-gpu
最后测试一下tensorflow:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')