ChunML / ssd-tf2

A super clean implementation of SSD (Single Shot MultiBox Detector) made possible by Tensorflow 2.0
MIT License
115 stars 54 forks source link

error #14

Open jiollos opened 4 years ago

jiollos commented 4 years ago

I used my own dataset and have this problem:

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. [Op:Conv2D] The program is exiting...

my enviroment are python 3.7.4, tensorflow 2.1.0, cuda 10.1, cudnn 7

ChunML commented 4 years ago

are you using RTX20X0 GPU?