报错
两段下面这个信息
UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
解决方案:
代码前面是这样就突然可以跑了,我的版本是CUDA11.1+tensorflow-gpu2.6.0
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''
config1=tf.compat.v1.ConfigProto()
config1.gpu_options.allow_growth = True
tf.compat.v1.disable_eager_execution()
sess=tf.compat.v1.Session(config=config1)
#print(sess.graph)
graph = sess.graph
gpu_devices = tf.config.experimental.list_physical_devices("GPU")
for device in gpu_devices:
tf.config.experimental.set_memory_growth(device, True)
后来又遇到一次,有用的应该是这两句话:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''