Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion dev-requirements.in
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ pytest-icdiff
tensorflow<=2.15.1; python_version<'3.12'
# Newer versions of torch bring in nvidia dependencies that are not present in windows, so
# we put this constraint while we do not have per-environment requirements files
torch<=1.12.1; python_version<'3.11'
torch<=2.8.0; python_version<'3.11'
# pytorch 2 supports python 3.11
# pytorch 2 does not support 3.12 yet: https://github.com/pytorch/pytorch/issues/110436
torch; python_version<'3.12'
Expand Down
55 changes: 53 additions & 2 deletions dev-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,7 @@ executing==2.0.1
filelock==3.20.3
# via
# snowflake-connector-python
# torch
# virtualenv
flatbuffers==25.12.19
# via tensorflow
Expand All @@ -151,6 +152,7 @@ fsspec==2024.5.0
# flytekit
# gcsfs
# s3fs
# torch
gast==0.7.0
# via tensorflow
gcsfs==2024.5.0
Expand Down Expand Up @@ -265,6 +267,8 @@ jeepney==0.8.0
# via
# keyring
# secretstorage
jinja2==3.1.6
# via torch
jmespath==1.0.1
# via botocore
joblib==1.4.2
Expand Down Expand Up @@ -301,7 +305,9 @@ markdown-it-py==3.0.0
# flytekit
# rich
markupsafe==3.0.3
# via werkzeug
# via
# jinja2
# werkzeug
marshmallow==3.21.2
# via
# dataclasses-json
Expand Down Expand Up @@ -329,6 +335,8 @@ more-itertools==10.2.0
# via
# jaraco-classes
# jaraco-functools
mpmath==1.3.0
# via sympy
msal==1.28.0
# via
# azure-datalake-store
Expand All @@ -350,6 +358,8 @@ mypy-extensions==1.0.0
# typing-inspect
nest-asyncio==1.6.0
# via ipykernel
networkx==3.4.2
# via torch
nodeenv==1.9.0
# via pre-commit
numpy==1.26.4
Expand All @@ -363,6 +373,43 @@ numpy==1.26.4
# scipy
# tensorboard
# tensorflow
nvidia-cublas-cu12==12.8.4.1
# via
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
# torch
nvidia-cuda-cupti-cu12==12.8.90
# via torch
nvidia-cuda-nvrtc-cu12==12.8.93
# via torch
nvidia-cuda-runtime-cu12==12.8.90
# via torch
nvidia-cudnn-cu12==9.10.2.21
# via torch
nvidia-cufft-cu12==11.3.3.83
# via torch
nvidia-cufile-cu12==1.13.1.3
# via torch
nvidia-curand-cu12==10.3.9.90
# via torch
nvidia-cusolver-cu12==11.7.3.90
# via torch
nvidia-cusparse-cu12==12.5.8.93
# via
# nvidia-cusolver-cu12
# torch
nvidia-cusparselt-cu12==0.7.1
# via torch
nvidia-nccl-cu12==2.27.3
# via torch
nvidia-nvjitlink-cu12==12.8.93
# via
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
# torch
nvidia-nvtx-cu12==12.8.90
# via torch
oauthlib==3.2.2
# via
# kubernetes
Expand Down Expand Up @@ -569,6 +616,8 @@ stack-data==0.6.3
# via ipython
statsd==3.3.0
# via flytekit
sympy==1.14.0
# via torch
tensorboard==2.15.2
# via tensorflow
tensorboard-data-server==0.7.2
Expand All @@ -592,7 +641,7 @@ tomli==2.4.0
# setuptools-scm
tomlkit==0.13.2
# via snowflake-connector-python
torch==1.12.1 ; python_version < "3.12"
torch==2.8.0 ; python_version < "3.12"
# via -r dev-requirements.in
tornado==6.4.1
# via
Expand All @@ -606,6 +655,8 @@ traitlets==5.14.3
# jupyter-client
# jupyter-core
# matplotlib-inline
triton==3.4.0
# via torch
types-cachetools==6.2.0.20251022
# via -r dev-requirements.in
types-croniter==2.0.0.20240423
Expand Down
Loading