TensorFlow杂乱的笔记
程序员文章站
2022-06-16 22:54:11
...
2018/5/8
tensorflow 分配显存的问题
如果程序中不指定,tensorflow 默认分配全部显存,指定分配内存的代码如下:
import tensorflow as tf
config = tf.ConfigProto()
# 设置每个进程 GPU 最大占用量
config.gpu_options.per_process_gpu_memory_fraction = 0.5
# 程序按需申请显存
config.gpu_options.allow_growth = Ture
# Session 中设置配置
with tf.Session(config=config) as sess:
...
tf.gather()函数
x = tf.range(4,14)
# 索引
idx = [1,2,3,6,7,1,2]
# 按索引挑出对应的值
result = tf.gather(x, idx)
with tf.Session as sess:
print sess.run(result)
# 结果
[5 6 7 10 11 5 6]
上一篇: tf.pad()简单使用
推荐阅读
-
和大家分享一个php笔记的网站。
-
笔记本维护心得 保养笔记本电脑的十一点注意事项
-
制作python版本的类CIFAR10数据集.Tensorflow
-
用VirtualBox构建MySQL测试环境的笔记_MySQL
-
Java开发笔记(一百零七)URL地址的组成格式
-
Web前端新人笔记之height、min-height的区别_html/css_WEB-ITnose
-
php学习笔记 面向对象中[接口]与[多态性]的应用
-
荐 java生成图片,自己踩过的那些坑。java.awt的学习笔记!!!
-
PHP中的cURL请求及示例学习笔记_PHP教程
-
Mysql存储过程学习笔记--建立简单的存储过程_MySQL