Thank you @8bitmp3 for the reply. I already run plenty of tensorflow models on my computer
This model is running on a Ubuntu 20.10
The Graphic card is a GeForce GTX1060, driver Version: 460.80
CUDA Version: 11.2
cuDNN: 7.6.5