ubantu16.04 tensorflow2.0 cuda10.1 环境下出现:Failed to get convolution algorithm.

ubantu16.04 tensorflow2.0 cuda10.1 环境下出现:Failed to get convolution algorithm.

经过排查发现是因为可能是显卡内存溢出导致的错误在代码中添加如下代码即可避免:

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)

http://www.bubuko.com/infodetail-3235806.html
https://blog.csdn.net/qq_41868689/article/details/98503069

你可能感兴趣的:(ubantu16.04 tensorflow2.0 cuda10.1 环境下出现:Failed to get convolution algorithm.)