win10安装Tensorflowhtml
前提:python
肯定你的显卡是否支持cuda,以及cuda版本。利用NVIDIA控制面板查看,具体请百度。编程
Python3.5 以上。windows
安装的是tensorflow的gpu版本网络
安装:dom
以前,TensorFlow还不支持Window系统,虽然能够安装,可是极其麻烦。Google在2016年11月在开发者博客中宣布新版本 TensorFlow0.12,增长了对Window系统的支持。测试
安装前准备:spa
TensorFlow有两个版本CPU和GPU版本。GPU版本须要安装CUDA和cuDNN的支持,CPU版本直接安装便可。可是GPU图像计算强于cpu版,命令行
1.pip升级orm
首先查看电脑是否支持CUDA。确保你的Python版本是3.5 64位。确保你有稳定的网络链接。确保你的pip版本 >= 8.1。用 pip -V 查看当前 pip 版本,用 python -m pip install -U pip 升级pip 。
而后能够开始下载 CUDA 和 cuDNN 的安装包了,注意版本号,个人版本分别是 Ccuda_8.0.44_win10.exe和 cudnn-8.0-windows10-x64-v6.0。
2.cuda安装
本人的机器支持GPU,因此我安装的是GPU版本。
安装cuda https://developer.nvidia.com/cuda-downloads
将下载的文件解压,运行安装程序,安装过程屏幕可能会闪烁,安装时间有点长。安装以后系统变量会自动为你添加上。
测试一下是否安装成功,命令行输入nvcc –V ,看到版本信息就表示安装成功了。
3.cudnn安装
安装cuDNN 必须和cuda版本对应 https://developer.nvidia.com/cudnn 须要填写一些问卷
cuDNN是压缩包,解压以后放在须要存放的位置。须要的位置是指和cuda对应的文件夹,好比:将cuDNN文件夹bin的文件复制到cuda相应的文件夹bin下。其余的文件夹相似。
4.配置环境变量path
import tensorflow as tf
import numpy as np
import os
#忽略python的警告信息
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# 使用 NumPy 生成假数据(phony data), 总共 100 个点.
x_data = np.float32(np.random.rand(2, 100)) # 随机输入
y_data = np.dot([0.100, 0.200], x_data) + 0.300
# 构造一个线性模型
#
b = tf.Variable(tf.zeros([1]))
W = tf.Variable(tf.random_uniform([1, 2], -1.0, 1.0))
y = tf.matmul(W, x_data) + b
# 最小化方差
loss = tf.reduce_mean(tf.square(y - y_data))
optimizer = tf.train.GradientDescentOptimizer(0.5)
train = optimizer.minimize(loss)
# 初始化变量
init =tf.global_variables_initializer()
# 启动图 (graph)
sess = tf.Session()
sess.run(init)
# 拟合平面
for step in range(0, 201):
sess.run(train)
if step % 20 == 0:
print (step, sess.run(W), sess.run(b))