最近更新时间:2024-03-01 13:26:16
网络良好时,可直接指令安装:
pip3 install --upgrade tensorflow-gpu
进行tensorflow 1.5.0下载或复制下载地址 https://github.com/tensorflow/tensorflow
下载对应 GPU版本,这里我们选择python 3.5 版本,得到文件【tf_nightly_gpu-1.head-cp35-cp35m-linux_x86_64.whl】。
输入以下指令,进行安装。
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社区文档。
纯净模式