之前在ubuntu安装tensorflow走过很多弯路,也重装过很多次系统,今天在windows下安装时特意请求大神支援,一次性通过,现分享经验如下。以下安装包链接:链接: https://pan.baidu.com/s/1z3fr8kH3aztMEzwtApyRTg 提取码: 6c2a
1.安装anaconda(直接双击可执行文件即可)
2.安装tensorflow(切换到该二进制文件所在目录)
https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.8.0/py27/GPU/cuda92cudnn71sse2(windows下的二进制包)
在程序搜索界面找到Anaconda Prompt
输入安装命令:pip install tensorflow-gpu==1.8.0
python
import tensorflow as tf
会报错,接着安装相应配置。
3.配置GPU
(1)装相应版本NVIDIA显卡,去tensorflow官网查看,我的电脑自带显卡满足tensorflow安装要求。
https://tensorflow.google.cn/
(2) 装cuda9.2(双击可执行文件)(我安装在D盘的CUDA文件夹下)
(3)装cudnn9.2
解压后,将bin(头文件),include(可执行文件),lib(库文件)复制到CUDA根目录下的一个文件夹中,我的文件夹取名为cudann。
(4)需要配置环境变量的文件
1)CUDA下的bin
2)CUDA下的extras中的CUPTI中的libx64
3) cudann中的bin文件
(前两个都是cuda的环境变量,第三个是cudnn的环境变量)
(5)配置环境变量
“Add the CUDA, and cuDNN installation directories to the %PATH% environmental variable.
For example, if the CUDA Toolkit is installed to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
and cuDNN to C:\tools\cuda, update your %PATH% to match:
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64;%PATH%
SET PATH=C:\tools\cuda\bin;%PATH%”
根据实际情况输入自己的环境变量路径,千万不要弄错路径。
方法一:在电脑左下角搜索程序处找到 windows系统>命令提示符(右键)>更多>以管理员身份运行。
输入三个“SETPATH"命令即可
方法二:在控制面板更改环境变量。
我的设置路径如下:
4.测试
安装过程完成,再次在AnacondaPrompt命令窗口输入:python ,import tensorflow
接着进行测试:
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print sess.run(c)
将以上内容保存为 test.py文件放在桌面,切换到桌面,运行python test.py,结果如下:
出现GPU信息,tensorflow安装成功!
希望大家少走弯路!Good Luck! :)