安装完成以后,能够import tensorflow as tf没有问题,但是执行sess=tf.Session()出现问题

原因:本机有两块GPU
解决方案:
加三句
config = tf.ConfigProto()
config.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1
config.gpu_options.visible_device_list="0"
并且sess=tf.Session(config=config)


本文介绍了解决在配置有双GPU的机器上运行TensorFlow时遇到的问题,通过设置ConfigProto参数,指定可见设备列表为特定GPU,解决了sess=tf.Session()执行时的错误。

1766

被折叠的 条评论
为什么被折叠?



