How to write task definition based on different enviroments - dev, staging & prod

Question

I have an ECS service that is running a task with one container each. Challenge is here is that I have different compute configuration based on environment.

For example in dev environment I intend to use .25cpu with .5MB memory but in production I would want to provision it 2cpu and 4GB Memory.

My thought process for the above challenge:

1. Create separate task family for each service with separate configurations and container images. For ex: my-service-dev & my-service-prod
2. Create a standard template and sed service name to update values

What can be the most scalable way to tackle above challenge ?