这个问题可能是由于GPU资源不足导致的。您可以尝试减小batch size或者增加GPU的内存限制。另外,您可以尝试重新安装tensorflow或者升级到最新版本来解决问题。以下是可能有用的代码示例:
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 可以根据GPU内存大小设置
sess = tf.Session(config=config)
您也可以尝试使用以下代码来解决:
import tensorflow as tf
with tf.device('/cpu:0'):
# 代码放在这里
这将使您的代码在CPU上运行,而不是GPU上。希望这可以帮助您解决问题。
你这个是jupyter?是的话重启kernel看下,顺便看下后台GPU被谁占用了,直接先干掉。类似一些动态桌面之类的吃GPU也很多,像wallpaper engine里面那个宇航员的就很占GPU