failed call to cuInit: CUDA_ERROR_NO_DEVICE

网上找了下相关问题:有两种解决方案

方案一 代码中添加:

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"

方案二

with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:

推荐方案1,因为方案二会打印一堆gpu信息

猜你喜欢

转载自blog.csdn.net/liuxiaodong400/article/details/81124251