Docker image creation for DLtrain