全部文档
当前文档

暂无内容

如果没有找到您期望的内容,请尝试其他搜索词

文档中心

安装TensorFlow指南

最近更新时间:2024-03-01 13:26:16

安装TensorFlow

网络安装

网络良好时,可直接指令安装:

pip3 install --upgrade tensorflow-gpu

源码安装

  1. 进行tensorflow 1.5.0下载或复制下载地址 https://github.com/tensorflow/tensorflow

  2. 下载对应 GPU版本,这里我们选择python 3.5 版本,得到文件【tf_nightly_gpu-1.head-cp35-cp35m-linux_x86_64.whl】。

    tensor.png
  3. 输入以下指令,进行安装。

    pip3 install tf_nightly_gpu-1.head-cp35-cp35m-linux_x86_64.whl

验证环境

TensorFlow安装完成后,执行python脚本,根据输出结果,确认是否有异常。Python代码如下:

import tensorflow as tf
import numpy as np
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.initialize_all_variables()
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))*

使用以上代码写入python 文件,Python 指向文件运行即可,如果运行成功则结果如下:

验证环境.jpg

更加详细的安装流程,可以参考tensorflow社区文档

文档导读
纯净模式常规模式

纯净模式

点击可全屏预览文档内容
文档反馈