TensorFlow Action（开山使用篇）-优快云博客

本文详细介绍如何使用pip安装TensorFlow CPU版和GPU版，并提供安装验证示例代码。此外，还介绍了如何指定GPU资源使用比例及配置TensorFlow运行会话。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、TensorFlow安装：

　　使用pip install tensorflow安装CPU版；

　　或使用pip install tensorflow-gpu==1.2.1指定版本安装GPU版。

2、TensorFlow测试样例：

import tensorflow as tf
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([3.0, 4.0], name="b")
add = a + b
sess = tf.Session()
sess.run(add)

　　使用python的import操作加载TensorFlow；

　　使用tf.constant函数定义两个常量tensor，并将两个tensor相加；

　　使用tf.Session函数生成一个会话（session）并通过这个session来计算结果；

　　以上若运行无误，则说明TensorFlow安装成功。

3、TensorFlow指定GPU占用示例：

import os
import tensorflow as tf
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
config = tf.ConfigProto()
# 设置定量的GPU使用量,占用GPU50%的显存
# config.gpu_options.per_process_gpu_memory_fraction = 0.5
# 设置最小的GPU使用量，自动增长：按需求分配显存
config.gpu_options.allow_growth = True
# 打印详细日志信息
config.log_device_placement = True
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([3.0, 4.0], name="b")
add = a + b
with tf.Session(config=config) as sess:
    sess.run(add)