gpu Cannot assign a device for operation Operation was explicitly assigned to
换成GPU 之后跑resnet跑不起来。。。。 之前好像训练都忘了用GPU了,并且
把
sess = tf.Session()
改成
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))
gpu Cannot assign a device for operation Operation was explicitly assigned to
换成GPU 之后跑resnet跑不起来。。。。 之前好像训练都忘了用GPU了,并且
把
sess = tf.Session()
改成
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))