网上找了下相关问题:有两种解决方案
方案一 代码中添加:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
方案二
with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
推荐方案1,因为方案二会打印一堆gpu信息
网上找了下相关问题:有两种解决方案
方案一 代码中添加:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
方案二
with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
推荐方案1,因为方案二会打印一堆gpu信息