我想用GPU版本来加速训练，请问需要在create

我想用GPU版本来加速训练，请问需要在create_model中做哪些改动？

来源：7-14 计算图实现

MacWorld2017

2019-02-20

图片描述

写回答

1回答

正十七

2019-02-24

同学您好，

不需要在create_model里做什么，需要在session里添加：

config = tf.ConfigProto(
    log_device_placement=True)
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
    ...

log_device_placement会记录下每个变量存在什么位置，gpu还是cpu上，还有本程序并非独占gpu(config.gpu_options.allow_growth = True)

另外，需要配置好gpu环境，主要是安装cuda和cudnn： https://blog.csdn.net/u014595019/article/details/53732015

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1