tensorflow1.0 windows下安装
Posted 不要做程序员的小松鼠
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow1.0 windows下安装相关的知识,希望对你有一定的参考价值。
主要步骤 17.03.22
- 安装python3.5 只能是3.5 64位
- 安装cuda8.0 只能是8.0 (显卡计算能力3.0以上)
- cuDnn
- 安装tensorflow
- 测试
-安装Python
双击安装就行了
安装前会让选择添加环境变量
安装完用cmd 测试一下即可 python -V 查看版本,或者pip -V
如果找不到命令,就自己将安装目录添加到环境变量Path
-安装cuda8.0 cuDnn
官网下载,双击安装就行了
之后运行python脚本,如果提示cudnn相关dll找不到,就将cuDnn里内容分别复制到cuda的安装目标:默认为C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v8.0
-安装tensorflow
方法一:直接 pip install tensorflow-gpu
方法二:下载whl文件的方式,直接用pip安装whl文件
pip install 路径\\tensorflow_gpu-1.0.0-cp35-cp35m-win_amd64.whl
ps:建议先把numpy等安装好,否则容易下载超时
安装numpy:下载whl文件,pip install 安装
测试
写个python脚本测试下
测试一:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
测试二:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
#Creates a graph.
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)
#Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
#Runs the op.
print(sess.run(c))
注意python2和python3的语法区别
出现warning,暂时忽略。
下载链接
- python-3.5.3-amd64.exe
- cudnn-7.5-windows7-x64-v5.0-ga
- numpy-1.12.1-cp35-none-win_amd64.whl
- tensorflow_gpu-1.0.0-cp35-cp35m-win_amd64.whl
只教安装不教卸载都是耍流氓
卸载:pip uninstall tensorflow-gpu
欢迎留言讨论。
以上是关于tensorflow1.0 windows下安装的主要内容,如果未能解决你的问题,请参考以下文章
初用Linux, 安装Ubuntu16.04+NVIDIA387+CUDA8.0+cudnn5.1+TensorFlow1.0.1
TensorFlow 1.0 在 Windows 上看不到 GPU(但 Theano 有)