tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况

在跑的时候可以让加些选项: with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)): 其中allow_soft_placement能让tensorflow遇到无法用GPU跑的数据时,自动切换成CPU进行。 log_device_placement则记录一些日志。

原文地址:https://www.cnblogs.com/tangmiao/p/8297487.html