Is it possible to run lamba container image that wrap another docker image?

I have a docker image (let call it image A), it is a ML docker image, with nvidia runtime with size is around 6GB. To run this image, i need to pass argument to it (python main.py -....) Is it possible to wrap image A into a lambda container runtime and use lambda function to execute the same logic (pass argument to A)? (I cannot change any code of image A).