这种问题是调用GPU出现问题,目前我发现的有两种解决方案:
一.调用代码
在session前加上
os.environ[“CUDA_VISIBLE_DEVICES”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0’ #use GPU with ID=0
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
session = tf.Session(config=config)
二.用指令清除缓存
sudo rm -rf ~/.nv/
或者重启