解決辦法
用tensorflow的時候都用下面三行:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
用tensorflow的時候都用下面三行:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
1、Verify You Have a CUDA-Capable GPU lspci | grep -i nvidia 2、Verify Yo