Skip to content
Open
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
65 changes: 52 additions & 13 deletions requirements/base.txt
Original file line number Diff line number Diff line change
Expand Up @@ -60,18 +60,19 @@ cryptography==42.0.8
cycler==0.12.1
# via matplotlib
determined==0.26.4
# via -r requirements/base.in
# via -r base.in
docker[ssh]==7.1.0
# via determined
docker-pycreds==0.4.0
# via wandb
einops==0.8.0
# via -r requirements/base.in
# via -r base.in
filelock==3.15.3
# via
# determined
# huggingface-hub
# torch
# triton
# yogadl
fonttools==4.53.0
# via matplotlib
Expand Down Expand Up @@ -123,6 +124,8 @@ huggingface-hub==0.23.4
# via tokenizers
idna==3.7
# via requests
importlib-resources==6.5.2
# via matplotlib
isodate==0.6.1
# via azure-storage-blob
jinja2==3.1.4
Expand All @@ -132,7 +135,7 @@ jmespath==1.0.1
# boto3
# botocore
jsonlines==4.0.0
# via -r requirements/base.in
# via -r base.in
kiwisolver==1.4.5
# via matplotlib
lmdb==1.4.1
Expand All @@ -149,14 +152,45 @@ monotonic==1.6
# via analytics-python
mpmath==1.3.0
# via sympy
networkx==3.3
networkx==3.2.1
# via torch
numpy==2.0.0
# via
# -r requirements/base.in
# -r base.in
# contourpy
# determined
# matplotlib
nvidia-cublas-cu12==12.1.3.1
# via
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
# torch
nvidia-cuda-cupti-cu12==12.1.105
# via torch
nvidia-cuda-nvrtc-cu12==12.1.105
# via torch
nvidia-cuda-runtime-cu12==12.1.105
# via torch
nvidia-cudnn-cu12==8.9.2.26
# via torch
nvidia-cufft-cu12==11.0.2.54
# via torch
nvidia-curand-cu12==10.3.2.106
# via torch
nvidia-cusolver-cu12==11.4.5.107
# via torch
nvidia-cusparse-cu12==12.1.0.106
# via
# nvidia-cusolver-cu12
# torch
nvidia-nccl-cu12==2.18.1
# via torch
nvidia-nvjitlink-cu12==12.9.86
# via
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
# via torch
packaging==24.1
# via
# determined
Expand Down Expand Up @@ -193,7 +227,7 @@ pyasn1-modules==0.4.0
pycparser==2.22
# via cffi
pydantic==2.7.4
# via -r requirements/base.in
# via -r base.in
pydantic-core==2.18.4
# via pydantic
pynacl==1.5.0
Expand All @@ -206,7 +240,7 @@ pyparsing==3.1.2
# matplotlib
python-dateutil==2.9.0.post0
# via
# -r requirements/base.in
# -r base.in
# analytics-python
# botocore
# determined
Expand Down Expand Up @@ -237,7 +271,7 @@ ruamel-yaml-clib==0.2.8
# via ruamel-yaml
s3transfer==0.10.1
# via boto3
sentry-sdk==2.6.0
sentry-sdk==2.8.0
# via wandb
setproctitle==1.3.3
# via wandb
Expand All @@ -258,14 +292,16 @@ tabulate==0.9.0
termcolor==2.4.0
# via determined
tokenizers==0.19.1
# via -r requirements/base.in
# via -r base.in
torch==2.1.2
# via -r requirements/base.in
# via -r base.in
tqdm==4.66.4
# via
# -r requirements/base.in
# -r base.in
# determined
# huggingface-hub
triton==2.1.0
# via torch
typing-extensions==4.12.2
# via
# azure-core
Expand All @@ -274,20 +310,23 @@ typing-extensions==4.12.2
# pydantic
# pydantic-core
# torch
# wandb
uritemplate==4.1.1
# via google-api-python-client
urllib3==2.2.2
urllib3==1.26.20
# via
# botocore
# docker
# requests
# sentry-sdk
wandb==0.17.2
# via -r requirements/base.in
# via -r base.in
websockets==12.0
# via yogadl
yogadl==0.1.4
# via determined
zipp==3.23.0
# via importlib-resources

# The following packages are considered to be unsafe in a requirements file:
# setuptools