Repository navigation
How do pin an ML job to a specific runtime image version and python version? #230
Description
Hi!
I am using the decorator below to start an ML job.
@jobs.remote(
compute_pool=COMPUTE_POOL,
stage_name=JOB_STAGE,
database=ML_DATABASE_ENV,
schema=JOBS_SCHEMA,
runtime_environment='???',
)
I am trying to pin the runtime version to specific container runtime, and python version. But both doesn't seem to be possible?
I have tried 2.6.0, which resolves to Python 3.10 (my project uses 3.12).
I have also tried with {"pythonVersion": "3.12"}, which gives me the latest runtime version, with python 3.12. Problem is that latest might change, including the installed package version which I'd like to avoid.
Right now I have reverted to supplying the complete image name, but that will only work as long as we are using CPU compute pools: "/snowflake/images/snowflake_images/container_runtime/cpu_x86_64:2.6.0-py312".
How can I pin the environment to a container version and python version at the same time?
Activity
sfc-gh-kunli commentedon Jun 12, 2026
See answer below.
sfc-gh-ajiang commentedon Jun 17, 2026
We support users to specify the full image url or image tag at
runtime_environmentthe image tag follow this format
xxx-py{version}, for examplepython 3.10 - you could use the image tag
2.6.0or2.6.0-py310python 3.11 - you could use the image tag
2.6.0-py311python 3.12 - you could use the image tag
2.6.0-py312