@Kamal Tanwar Thanks for the question. Can you please add more details about the docker container that you are trying.
Please follow the document to train a model by using custom docker image. The base images are available on Github.
You can also use these Docker images as base images for your custom Azure ML Environments. If you specify any conda dependencies in your Environment, the extra dependencies are installed on top of the dependencies in the Docker image.
Please follow Prebuilt Docker images for inference (before Inference-optimized curated images). This provides with a faster and simpler deployment experience. These prebuilt Docker Images come with popular machine learning frameworks and python packages.