Snowflake Container Runtime GPU Version 2.9 (Latest)

Added

  • Per-job tuning of the file-level disk cache via a file_cache_config dict, accepted alongside use_file_cache on every stage datasource, create_stage_filesystem, SFStageFileSystem and RayDataIngester — e.g. SFStageShardDataSource(..., use_file_cache=True, file_cache_config={"max_disk_pct": 70}). Knobs: max_disk_pct (cap the cache at that percent of disk, overriding MLRS_FILE_CACHE_MAX_DISK_PCT), evict_interval_sec (how often least-recently-used eviction may scan; default 30s), cache_dir (absolute path replacing $TMPDIR as the cache root, to put the cache on a different filesystem) and disk_subdir (the path within that root). Omitted knobs keep their current behaviour, and an unrecognised or malformed knob is rejected up front rather than silently ignored. Like use_file_cache, the settings reach every Ray worker, so they apply cluster-wide for the job; the cache location is fixed once the cache is in use, so set it before the first cached read.

The following lists the packages available for each Python version of GPU version 2.9.

GPU Container Runtime Python 3.10 Version 2.9 has the following packages:

PackageVersion
absl-py2.5.0
accelerate1.14.0
aiobotocore2.26.0
aiohappyeyeballs2.7.1
aiohttp3.14.3
aiohttp-cors0.8.1
aioitertools0.13.0
aiosignal1.4.0
altair5.5.0
annotated-doc0.0.5
annotated-types0.8.0
anyio4.14.2
anywidget0.11.0
appdirs1.4.4
argon2-cffi25.1.0
argon2-cffi-bindings25.1.0
arrow1.4.0
arviz0.23.4
asn1crypto1.5.1
asttokens3.0.2
async-lru2.3.0
async-timeout5.0.1
attrs26.1.0
babel2.18.0
bayesian-optimization1.5.1
beautifulsoup44.15.0
bleach6.4.0
blinker1.9.0
boto31.41.5
botocore1.41.5
cachetools7.1.7
causalpy0.5.0
certifi2026.7.22
cffi2.1.1
charset-normalizer3.5.1
click8.2.1
clikit0.6.2
cloudpickle3.1.1
cmdstanpy1.3.0
colorama0.4.6
colorful0.5.8
comm0.2.3
cons0.4.7
contourpy1.3.2
crashtest0.3.1
cryptography50.0.0
cuda-bindings12.9.7
cuda-core0.7.0
cuda-pathfinder1.6.0
cuda-python12.9.7
cuda-toolkit12.9.1
cudf-cu1226.2.1
cuml-cu1226.2.0
cupy-cuda12x14.1.1
cycler0.12.1
datasets5.0.1
datasketches5.2.0
debugpy1.8.21
decorator5.3.1
deepdiff8.6.2
defusedxml0.7.1
dill0.4.1
distlib0.4.3
etuples0.3.10
evaluate0.4.6
exceptiongroup1.3.1
executing2.2.1
faiss-gpu-cu121.14.1.post1
fastapi0.141.1
fastjsonschema2.22.2
filelock3.32.3
flaml2.6.0
flask3.1.3
fonttools4.63.0
fqdn1.5.1
frozenlist1.8.0
fsspec2025.12.0
gcsfs2025.12.0
geojson3.3.0
google-api-core2.34.0
google-auth2.56.3
google-auth-oauthlib1.4.0
google-cloud-core2.6.1
google-cloud-storage3.13.1
google-cloud-storage-control1.13.0
google-crc32c1.8.0
google-resumable-media2.10.1
googleapis-common-protos1.75.1
graphviz0.21
grpc-google-iam-v10.14.5
grpcio1.78.0
grpcio-status1.78.0
gunicorn25.0.3
h110.16.0
h24.4.1
h5netcdf1.8.1
h5py3.16.0
hf-xet1.6.0
holidays0.103
hpack4.2.0
httpcore1.0.9
httpstan4.13.0
httpx0.28.1
huggingface-hub1.16.1
hyperframe6.1.0
idna3.18
importlib-resources6.5.2
inflection0.5.1
ipykernel7.3.0
ipython8.39.0
ipywidgets8.1.8
isoduration20.11.0
itsdangerous2.2.0
jedi0.20.0
jinja23.1.6
jmespath1.1.0
joblib1.5.3
jpype11.7.1
json50.15.0
jsonpointer3.1.1
jsonschema4.26.0
jsonschema-specifications2025.9.1
jupyter-builder1.2.2
jupyter-client8.9.1
jupyter-core5.9.1
jupyter-events0.12.1
jupyter-lsp2.3.1
jupyter-server2.20.0
jupyter-server-terminals0.5.4
jupyterlab4.6.3
jupyterlab-pygments0.3.0
jupyterlab-server2.28.0
jupyterlab-widgets3.0.16
kiwisolver1.5.0
lark1.3.1
libcudf-cu1226.2.1
libcuml-cu1226.2.0
libkvikio-cu1226.2.0
libraft-cu1226.2.0
librmm-cu1226.2.0
lightgbm4.7.0
lightgbm-ray0.1.9
llvmlite0.44.0
logical-unification0.4.7
markdown-it-py4.2.0
markupsafe3.0.3
marshmallow3.26.2
matplotlib3.10.9
matplotlib-inline0.2.2
mdurl0.1.2
minikanren1.0.5
mistune3.3.4
mlruntimes-service2.10.15
modin0.37.1
mpmath1.3.0
msgpack1.2.1
multidict6.7.1
multipledispatch1.0.0
multiprocess0.70.19
mypy-extensions1.1.0
narwhals2.24.0
nbclient0.11.0
nbconvert7.17.1
nbformat5.11.1
nest-asyncio21.7.2
networkx3.4.2
nltk3.10.3
nodeenv1.10.0
nodejs-wheel-binaries24.16.0
notebook7.6.2
notebook-shim0.2.4
numba0.61.2
numba-cuda0.22.2
numpy2.2.6
nvidia-cublas-cu1212.9.1.4
nvidia-cuda-cccl-cu1212.9.27
nvidia-cuda-cupti-cu1212.9.79
nvidia-cuda-nvcc-cu1212.9.86
nvidia-cuda-nvrtc-cu1212.9.86
nvidia-cuda-runtime-cu1212.9.79
nvidia-cudnn-cu129.10.2.21
nvidia-cufft-cu1211.4.1.4
nvidia-cufile-cu121.14.1.1
nvidia-curand-cu1210.3.10.19
nvidia-cusolver-cu1211.7.5.82
nvidia-cusparse-cu1212.5.10.65
nvidia-cusparselt-cu120.7.1
nvidia-libnvcomp-cu125.1.0.21
nvidia-ml-py13.610.43
nvidia-nccl-cu122.27.5
nvidia-nvjitlink-cu1212.9.86
nvidia-nvshmem-cu123.3.20
nvidia-nvtx-cu1212.9.79
nvtx0.2.16
oauthlib3.3.1
opencensus0.11.4
opencensus-context0.1.3
openfe0.0.12
opentelemetry-api1.44.0
opentelemetry-exporter-otlp-proto-common1.44.0
opentelemetry-exporter-otlp-proto-grpc1.44.0
opentelemetry-exporter-prometheus0.65b0
opentelemetry-proto1.44.0
opentelemetry-sdk1.44.0
opentelemetry-semantic-conventions0.65b0
orderly-set5.5.0
overrides7.7.0
packaging25.0
pandapower3.5.3
pandas2.3.3
pandera0.26.1
pandocfilters1.5.1
parso0.8.7
pastel0.2.1
patsy1.0.2
peft0.17.1
pexpect4.9.0
pillow12.3.0
platformdirs4.11.3
plotly6.9.0
polars1.43.2
polars-runtime-321.43.2
prometheus-client0.22.1
prompt-toolkit3.0.53
propcache0.5.2
prophet1.4.0
proto-plus1.28.3
protobuf6.33.6
psutil7.2.2
psygnal0.15.1
ptyprocess0.7.0
pure-eval0.2.3
py-spy0.4.2
py4j0.10.9.9
pyarrow24.0.0
pyasn10.6.4
pyasn1-modules0.4.2
pycparser3.0
pydantic2.13.4
pydantic-core2.46.4
pygments2.21.0
pyjwt2.13.0
pylev1.4.0
pylibcudf-cu1226.2.1
pylibraft-cu1226.2.0
pymc5.25.1
pyopenssl26.4.0
pyparsing3.3.2
pyright1.1.411
pysimdjson6.0.2
pystan3.10.0
pytensor2.31.7
python-dateutil2.9.0.post0
python-discovery1.5.2
python-json-logger4.2.0
pytimeparse1.1.8
pytz2026.3.post1
pyyaml6.0.3
pyzmq27.1.0
rapids-logger0.2.3
ray2.55.1
referencing0.37.0
regex2026.7.19
requests2.34.2
requests-oauthlib2.0.0
retrying1.4.2
rfc3339-validator0.1.4
rfc3986-validator0.1.1
rfc3987-syntax1.1.0
rich15.0.0
rmm-cu1226.2.0
rpds-py0.30.0
s3fs2025.12.0
s3transfer0.15.0
safetensors0.8.0
scikit-learn1.7.2
scipy1.15.3
seaborn0.13.2
send2trash2.1.0
sentencepiece0.2.2
setuptools84.0.0
shap0.49.1
shellingham1.5.4
six1.17.0
slicer0.0.8
smart-open8.0.1
snowflake1.13.1
snowflake-connector-python4.7.2
snowflake-core1.13.1
snowflake-legacy1.0.3
snowflake-ml-python1.51.0
snowflake-snowpark-python1.54.0
snowpark-connect1.38.0
snowpark-connect-deps-13.56.6
snowpark-connect-deps-23.56.6
snowpark-connect-deps-iceberg1.0.3
sortedcontainers2.4.0
soupsieve2.9.2
sqlglot30.17.0
sqlparse0.6.0
stack-data0.6.3
stanio0.5.1
starlette1.6.0
statsmodels0.14.6
sympy1.14.0
terminado0.18.1
threadpoolctl3.6.0
tinycss21.5.1
tokenizers0.22.2
tomli2.4.1
tomlkit0.15.1
toolz1.1.0
torch2.9.1+cu129
torchvision0.24.1+cu129
tornado6.5.8
tqdm4.70.0
traitlets5.16.1
transformers5.15.0
treelite4.7.0
triton3.5.1
typeguard4.6.0
typer0.27.1
typing-extensions4.16.0
typing-inspect0.9.0
typing-inspection0.4.4
tzdata2026.3
tzlocal5.4.4
uri-template1.3.0
urllib32.7.0
uvicorn0.52.3
virtualenv21.7.4
wcwidth0.8.2
webargs8.7.1
webcolors25.10.0
webencodings0.6.1
websocket-client1.9.0
werkzeug3.1.8
wheel0.48.0
widgetsnbextension4.0.15
wrapt1.17.3
xarray2025.6.1
xarray-einstats0.8.0
xgboost3.2.0
xgboost-ray0.1.19
xxhash4.0.1
yarl1.24.5