Tensorflow遇到的坑

Tensorflow遇到的坑

Scroll Down

报错 UnknownError: Failed to get convolution algorithm

QQ图片20200202045744
gpu改成按需分配

gpu = tf.config.experimental.list_physical_devices(device_type='GPU')[0]
tf.config.experimental.set_memory_growth(gpu, True)