tensorlayer / TensorLayer

Deep Learning and Reinforcement Learning Library for Scientists and Engineers
http://tensorlayerx.com
Other
7.34k stars 1.61k forks source link

请问能不能实现一机多GPU数据或者模型并行处理? #1031

Closed yonghuixu closed 5 years ago

yonghuixu commented 5 years ago

New Issue Checklist

Issue Description

[INSERT DESCRIPTION OF THE PROBLEM]

Reproducible Code

[INSERT CODE HERE]

# ======================================================== #
###### THIS CODE IS AN EXAMPLE, REPLACE WITH YOUR OWN ######
# ======================================================== #

import tensorflow as tf
import tensorlayer as tl

x = tf.placeholder(tf.float32, [None, 64])
net_in = tl.layers.InputLayer(x)

net = tl.layers.DenseLayer(net_in, n_units=25, act=tf.nn.relu, name='relu1')

print("Output Shape:", net.outputs.get_shape().as_list()) ### Output Shape: [None, 25]

# ======================================================== #
###### THIS CODE IS AN EXAMPLE, REPLACE WITH YOUR OWN ######
# ======================================================== #
zsdonghao commented 5 years ago

Horvord supports tensorlayer

yonghuixu commented 5 years ago

谢谢,我了解一下

---原始邮件--- 发件人: "Hao"notifications@github.com 发送时间: 2019年7月26日(星期五) 晚上6:02 收件人: "tensorlayer/tensorlayer"tensorlayer@noreply.github.com; 抄送: "YonghuiXu"2259949930@qq.com;"Author"author@noreply.github.com; 主题: Re: [tensorlayer/tensorlayer] 请问能不能实现一机多GPU数据或者模型并行处理? (#1031)

Horvord supports tensorlayer

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.