安装TensorFlow指南

最近更新时间:2020-10-23 18:42:08

安装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】。 安装TensorFlow指南
  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 指向文件运行即可,如果运行成功则结果如下: 安装TensorFlow指南

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

金山云,开启您的云计算之旅

免费注册