Dockerfile for Training Environment

[Raman1121]

Hello, thank you for the great work! Is it possible to provide us with the Dockerfile for setting up the training environment? I have seen the dockerfile for running inference and tried to adapt it for training but it is not functioning properly.

Thanks!