Rolling your own custom container docker image: how to create user defined command for local system GPU

Question

I would like to use my computer GPU to run the program. For that custom container has been created using: https://blog.softwaremill.com/setting-up-tensorflow-with-gpu-accelerati