From 6cdc40efb7f783839f941097065bbe7a29f6d822 Mon Sep 17 00:00:00 2001 From: "pierre.delaunay" Date: Tue, 4 Feb 2025 13:22:49 -0500 Subject: [PATCH 1/3] update pytorch to 2.6 --- .pin/constraints-cuda-torch.txt | 189 ++- .pin/constraints-rocm-torch.txt | 1061 +++++++++++++++++ benchmarks/brax/requirements.cuda.txt | 61 +- benchmarks/brax/requirements.rocm.txt | 189 +-- benchmarks/diffusion/requirements.cuda.txt | 55 +- benchmarks/diffusion/requirements.rocm.txt | 173 ++- benchmarks/dinov2/requirements.cuda.txt | 45 +- benchmarks/dinov2/requirements.rocm.txt | 111 +- benchmarks/flops/requirements.cuda.txt | 45 +- benchmarks/flops/requirements.rocm.txt | 89 +- benchmarks/geo_gnn/requirements-pre.cuda.txt | 37 +- benchmarks/geo_gnn/requirements-pre.rocm.txt | 29 +- benchmarks/geo_gnn/requirements.cuda.txt | 48 +- benchmarks/geo_gnn/requirements.rocm.txt | 137 ++- benchmarks/huggingface/requirements.cuda.txt | 47 +- benchmarks/huggingface/requirements.rocm.txt | 105 +- benchmarks/lightning/requirements.cuda.txt | 49 +- benchmarks/lightning/requirements.rocm.txt | 123 +- benchmarks/llama/requirements.cuda.txt | 51 +- benchmarks/llama/requirements.rocm.txt | 155 ++- benchmarks/llava/requirements.cuda.txt | 51 +- benchmarks/llava/requirements.rocm.txt | 153 ++- benchmarks/llm/requirements.cuda.txt | 185 ++- benchmarks/llm/requirements.rocm.txt | 327 +++-- benchmarks/purejaxrl/requirements.cuda.txt | 79 +- benchmarks/purejaxrl/requirements.rocm.txt | 337 +++--- benchmarks/recursiongfn/requirements.cuda.txt | 88 +- benchmarks/recursiongfn/requirements.rocm.txt | 256 ++-- benchmarks/rlhf/requirements.cuda.txt | 53 +- benchmarks/rlhf/requirements.rocm.txt | 180 ++- benchmarks/timm/requirements.cuda.txt | 47 +- benchmarks/timm/requirements.rocm.txt | 101 +- benchmarks/torchatari/requirements.cuda.txt | 59 +- benchmarks/torchatari/requirements.rocm.txt | 147 +-- benchmarks/torchvision/requirements.cuda.txt | 45 +- benchmarks/torchvision/requirements.rocm.txt | 89 +- .../torchvision_ddp/requirements.cuda.txt | 45 +- .../torchvision_ddp/requirements.rocm.txt | 89 +- benchmarks/vjepa/requirements.cuda.txt | 49 +- benchmarks/vjepa/requirements.rocm.txt | 135 ++- constraints/cuda.txt | 2 +- constraints/rocm.txt | 2 +- milabench/_version.py | 6 +- 43 files changed, 3432 insertions(+), 1892 deletions(-) create mode 100644 .pin/constraints-rocm-torch.txt diff --git a/.pin/constraints-cuda-torch.txt b/.pin/constraints-cuda-torch.txt index 7ab5ddd0..e180a723 100644 --- a/.pin/constraints-cuda-torch.txt +++ b/.pin/constraints-cuda-torch.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=.pin/constraints-cuda-torch.txt .pin/tmp-constraints.txt benchmarks/brax/requirements.in benchmarks/diffusion/requirements.in benchmarks/dinov2/requirements.in benchmarks/flops/requirements.in benchmarks/geo_gnn/requirements-pre.in benchmarks/geo_gnn/requirements.in benchmarks/huggingface/requirements.in benchmarks/lightning/requirements.in benchmarks/llama/requirements.in benchmarks/llava/requirements.in benchmarks/llm/requirements.in benchmarks/purejaxrl/requirements.in benchmarks/recursiongfn/requirements.in benchmarks/rlhf/requirements.in benchmarks/timm/requirements.in benchmarks/torchatari/requirements.in benchmarks/torchvision/requirements.in benchmarks/torchvision_ddp/requirements.in benchmarks/vjepa/requirements.in constraints/extra/torch.cuda.txt # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -16,6 +16,7 @@ absl-py==2.1.0 # chex # distrax # dm-env + # dm-tree # ml-collections # mujoco # mujoco-mjx @@ -58,11 +59,15 @@ asttokens==3.0.0 # via giving async-timeout==5.0.1 # via aiohttp -attrs==24.3.0 - # via aiohttp +attrs==25.1.0 + # via + # aiohttp + # dm-tree beartype==0.19.0 # via -r benchmarks/vjepa/requirements.in -black==24.10.0 +betterproto==2.0.0b6 + # via sigstore-protobuf-specs +black==25.1.0 # via navix blinker==1.9.0 # via flask @@ -72,7 +77,7 @@ blobfile==3.0.0 # torchtune blosc2==2.7.1 # via tables -botorch==0.12.0 +botorch==0.13.0 # via -r benchmarks/recursiongfn/requirements.in braceexpand==0.1.7 # via @@ -84,10 +89,12 @@ brax==0.12.1 # -r benchmarks/purejaxrl/requirements.in cantilever==0.1.0 # via -r benchmarks/torchatari/requirements.in -certifi==2024.12.14 +certifi==2025.1.31 # via # requests # sentry-sdk +cffi==1.17.1 + # via cryptography charset-normalizer==3.4.1 # via requests chex==0.1.88 @@ -113,6 +120,12 @@ codefind==0.1.7 # via ptera contourpy==1.3.1 # via matplotlib +cryptography==44.0.0 + # via + # model-signing + # pyopenssl + # rfc3161-client + # sigstore cvxopt==1.3.2 # via -r benchmarks/recursiongfn/requirements.in cycler==0.12.1 @@ -145,10 +158,12 @@ dm-env==1.6 # brax # envpool # rlax -dm-tree==0.1.8 +dm-tree==0.1.9 # via # dm-env # tensorflow-probability +dnspython==2.7.0 + # via email-validator docker-pycreds==0.4.0 # via wandb docstring-parser==0.16 @@ -159,6 +174,8 @@ einops==0.8.0 # via # -r benchmarks/rlhf/requirements.in # -r benchmarks/vjepa/requirements.in +email-validator==2.2.0 + # via pydantic envpool==0.8.4 # via -r benchmarks/torchatari/requirements.in etils[epath,epy]==1.11.0 @@ -172,7 +189,7 @@ evosax==0.1.6 # via -r benchmarks/purejaxrl/requirements.in exceptiongroup==1.2.2 # via pytest -executing==2.1.0 +executing==2.2.0 # via varname fairscale==0.4.13 # via @@ -189,7 +206,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton fire==0.7.0 # via # -r benchmarks/llama/requirements.in @@ -212,7 +228,7 @@ flax==0.10.2 # flashbax # gymnax # navix -fonttools==4.55.4 +fonttools==4.55.8 # via matplotlib frozenlist==1.5.0 # via @@ -243,14 +259,16 @@ giving==0.4.3 # voir glfw==2.8.0 # via mujoco -gpytorch==1.13 +gpytorch==1.14 # via # -r benchmarks/recursiongfn/requirements.in # botorch -grpcio==1.69.0 +grpcio==1.70.0 # via # brax # tensorboard +grpclib==0.4.7 + # via betterproto gym==0.26.2 # via # -r benchmarks/torchatari/requirements.in @@ -267,9 +285,13 @@ gymnax==0.0.8 # via # -c .pin/../constraints/cuda.txt # -r benchmarks/purejaxrl/requirements.in +h2==4.2.0 + # via grpclib hjson==3.1.0 # via argklass -huggingface-hub[hf-transfer]==0.27.1 +hpack==4.1.0 + # via h2 +huggingface-hub[hf-transfer]==0.28.1 # via # -r benchmarks/timm/requirements.in # accelerate @@ -281,25 +303,33 @@ huggingface-hub[hf-transfer]==0.27.1 # transformers humanize==4.11.0 # via orbax-checkpoint +hyperframe==6.1.0 + # via h2 +id==1.5.0 + # via sigstore idna==3.10 # via + # email-validator # requests # yarl importlib-metadata==8.6.1 # via diffusers -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # argklass # cantilever # etils + # sigstore # torchcompat +in-toto-attestation==0.9.3 + # via model-signing iniconfig==2.0.0 # via pytest iopath==0.1.10 # via # -r benchmarks/dinov2/requirements.in # fvcore -isort==5.13.2 +isort==6.0.0 # via pylint itsdangerous==2.2.0 # via flask @@ -339,7 +369,7 @@ jaxlib==0.5.0 # rlax jaxopt==0.8.3 # via brax -jaxtyping==0.2.19 +jaxtyping==0.2.37 # via # gpytorch # linear-operator @@ -351,18 +381,18 @@ jinja2==3.1.5 # torch-geometric joblib==1.4.2 # via scikit-learn -kagglehub==0.3.6 +kagglehub==0.3.7 # via torchtune kiwisolver==1.4.8 # via matplotlib lightning==2.5.0.post0 # via -r benchmarks/lightning/requirements.in -lightning-utilities==0.11.9 +lightning-utilities==0.12.0 # via # lightning # pytorch-lightning # torchmetrics -linear-operator==0.5.3 +linear-operator==0.6 # via # botorch # gpytorch @@ -394,9 +424,10 @@ ml-dtypes==0.5.1 # jax # jaxlib # tensorstore +model-signing==0.2.0 + # via kagglehub mpmath==1.3.0 # via - # botorch # gpytorch # linear-operator # sympy @@ -414,13 +445,16 @@ mujoco-mjx==3.2.7 multidict==6.1.0 # via # aiohttp + # grpclib # yarl multipledispatch==1.0.0 # via botorch multiprocess==0.70.16 # via datasets mypy-extensions==1.0.0 - # via black + # via + # black + # typing-inspect navix==0.7.0 # via -r benchmarks/purejaxrl/requirements.in ndindex==1.9.2 @@ -452,6 +486,7 @@ numpy==1.26.4 # diffusers # distrax # dm-env + # dm-tree # envpool # evosax # fairscale @@ -462,7 +497,6 @@ numpy==1.26.4 # jax # jaxlib # jaxopt - # jaxtyping # matplotlib # ml-dtypes # mujoco @@ -491,44 +525,46 @@ numpy==1.26.4 # trimesh # webdataset # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via torch +nvidia-ml-py==12.570.86 # via voir nvidia-nccl-cu12==2.21.5 # via @@ -537,9 +573,11 @@ nvidia-nccl-cu12==2.21.5 nvidia-nvjitlink-cu12==12.6.85 # via # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via torch omegaconf==2.3.0 # via @@ -560,7 +598,7 @@ optax==0.2.4 # flax optree==0.14.0 # via envpool -orbax-checkpoint==0.11.0 +orbax-checkpoint==0.11.1 # via # brax # flax @@ -610,6 +648,7 @@ platformdirs==4.3.6 # via # black # pylint + # sigstore # wandb pluggy==1.5.0 # via pytest @@ -621,6 +660,7 @@ propcache==0.2.1 # yarl protobuf==5.29.3 # via + # in-toto-attestation # orbax-checkpoint # tensorboard # tensorboardx @@ -642,26 +682,39 @@ pyarrow==19.0.0 # via # -r benchmarks/recursiongfn/requirements.in # datasets +pyasn1==0.6.1 + # via sigstore pycodestyle==2.12.1 # via flake8 +pycparser==2.22 + # via cffi pycryptodomex==3.21.0 # via blobfile -pydantic==2.10.5 - # via wandb +pydantic[email]==2.10.6 + # via + # sigstore + # sigstore-rekor-types + # wandb pydantic-core==2.27.2 # via pydantic pyflakes==3.2.0 # via flake8 pygments==2.19.1 # via rich -pylint==3.3.3 +pyjwt==2.10.1 + # via sigstore +pylint==3.3.4 # via navix pyopengl==3.1.9 # via mujoco +pyopenssl==25.0.0 + # via sigstore pyparsing==3.2.1 # via # matplotlib # torch-geometric +pyre-extensions==0.0.32 + # via botorch pyro-api==0.1.2 # via pyro-ppl pyro-ppl==1.9.1 @@ -672,13 +725,14 @@ pytest==8.3.4 # via navix python-dateutil==2.9.0.post0 # via + # betterproto # matplotlib # pandas pytinyrenderer==0.0.14 # via brax pytorch-lightning==2.5.0.post0 # via lightning -pytz==2024.2 +pytz==2025.1 # via pandas pyyaml==6.0.2 # via @@ -718,14 +772,22 @@ requests==2.32.3 # datasets # diffusers # huggingface-hub + # id # kagglehub + # sigstore # tiktoken # torch-geometric # transformers + # tuf # wandb +rfc3161-client==0.1.2 + # via sigstore +rfc8785==0.1.4 + # via sigstore rich==13.9.4 # via # flax + # sigstore # trl # tyro # voir @@ -758,6 +820,8 @@ scipy==1.15.1 # torch-sparse seaborn==0.13.2 # via gymnax +securesystemslib==1.2.0 + # via tuf sentencepiece==0.2.0 # via # -r benchmarks/llama/requirements.in @@ -770,6 +834,12 @@ setuptools-scm==8.1.0 # via navix shtab==1.7.1 # via tyro +sigstore==3.6.1 + # via model-signing +sigstore-protobuf-specs==0.3.2 + # via sigstore +sigstore-rekor-types==0.0.18 + # via sigstore simplejson==3.19.3 # via orbax-checkpoint six==1.17.0 @@ -811,7 +881,9 @@ termcolor==2.5.0 # fire # fvcore threadpoolctl==3.5.0 - # via scikit-learn + # via + # botorch + # scikit-learn tiktoken==0.8.0 # via torchtune timm==1.0.14 @@ -828,7 +900,7 @@ tomlkit==0.13.2 # via pylint toolz==1.0.0 # via chex -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../constraints/cuda.txt # -r benchmarks/brax/requirements.in @@ -877,7 +949,7 @@ torch-sparse==0.6.18+pt25cu121 # via # -r benchmarks/geo_gnn/requirements.in # -r benchmarks/recursiongfn/requirements.in -torchao==0.8.0 +torchao==0.8.0+cu126 # via -r benchmarks/llm/requirements.in torchcompat==1.1.4 # via @@ -894,7 +966,7 @@ torchmetrics==1.6.1 # pytorch-lightning torchtune==0.5.0 # via -r benchmarks/llm/requirements.in -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -r benchmarks/diffusion/requirements.in # -r benchmarks/dinov2/requirements.in @@ -922,7 +994,7 @@ tqdm==4.67.1 # torch-geometric # torchtune # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -r benchmarks/diffusion/requirements.in # -r benchmarks/huggingface/requirements.in @@ -931,18 +1003,18 @@ transformers==4.48.1 # -r benchmarks/llm/requirements.in # -r benchmarks/rlhf/requirements.in # trl -trimesh==4.6.0 +trimesh==4.6.1 # via # brax # mujoco-mjx -triton==3.1.0 +triton==3.2.0 # via torch -trl==0.13.0 +trl==0.14.0 # via -r benchmarks/rlhf/requirements.in +tuf==5.1.0 + # via sigstore typeguard==4.4.1 - # via - # jaxtyping - # tyro + # via tyro types-protobuf==5.29.1.20241207 # via envpool typing-extensions==4.12.2 @@ -959,15 +1031,17 @@ typing-extensions==4.12.2 # gymnasium # huggingface-hub # iopath - # jaxtyping # lightning # lightning-utilities + # model-signing # multidict # navix # optree # orbax-checkpoint # pydantic # pydantic-core + # pyopenssl + # pyre-extensions # pytorch-lightning # reactivex # rich @@ -975,9 +1049,12 @@ typing-extensions==4.12.2 # tables # torch # typeguard + # typing-inspect # tyro # wandb -tyro==0.9.12 +typing-inspect==0.9.0 + # via pyre-extensions +tyro==0.9.13 # via # -r benchmarks/torchatari/requirements.in # navix @@ -1011,7 +1088,9 @@ voir==0.2.19 # -r benchmarks/torchvision/requirements.in # -r benchmarks/torchvision_ddp/requirements.in # -r benchmarks/vjepa/requirements.in -wandb==0.19.4 +wadler-lindig==0.1.3 + # via jaxtyping +wandb==0.19.5 # via # -r benchmarks/recursiongfn/requirements.in # navix @@ -1021,7 +1100,9 @@ werkzeug==3.1.3 # via # flask # tensorboard -xformers==0.0.29.post1 +wrapt==1.17.2 + # via dm-tree +xformers==0.0.29.post2 # via # -r benchmarks/dinov2/requirements.in # -r constraints/extra/torch.cuda.txt diff --git a/.pin/constraints-rocm-torch.txt b/.pin/constraints-rocm-torch.txt new file mode 100644 index 00000000..2d0a0b8f --- /dev/null +++ b/.pin/constraints-rocm-torch.txt @@ -0,0 +1,1061 @@ +# +# This file is autogenerated by pip-compile with Python 3.10 +# by the following command: +# +# pip-compile --output-file=.pin/constraints-rocm-torch.txt .pin/tmp-constraints.txt benchmarks/brax/requirements.in benchmarks/diffusion/requirements.in benchmarks/dinov2/requirements.in benchmarks/flops/requirements.in benchmarks/geo_gnn/requirements-pre.in benchmarks/geo_gnn/requirements.in benchmarks/huggingface/requirements.in benchmarks/lightning/requirements.in benchmarks/llama/requirements.in benchmarks/llava/requirements.in benchmarks/llm/requirements.in benchmarks/purejaxrl/requirements.in benchmarks/recursiongfn/requirements.in benchmarks/rlhf/requirements.in benchmarks/timm/requirements.in benchmarks/torchatari/requirements.in benchmarks/torchvision/requirements.in benchmarks/torchvision_ddp/requirements.in benchmarks/vjepa/requirements.in constraints/extra/torch.rocm.txt +# +--extra-index-url https://pypi.ngc.nvidia.com +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 +--trusted-host pypi.ngc.nvidia.com + +absl-py==2.1.0 + # via + # brax + # chex + # distrax + # dm-env + # dm-tree + # ml-collections + # mujoco + # mujoco-mjx + # optax + # orbax-checkpoint + # rlax + # tensorboard + # tensorflow-probability +accelerate==1.3.0 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/llm/requirements.in + # -r benchmarks/rlhf/requirements.in + # diffusers + # trl +aiohappyeyeballs==2.4.4 + # via aiohttp +aiohttp==3.11.11 + # via + # datasets + # fsspec + # torch-geometric +aiosignal==1.3.2 + # via aiohttp +annotated-types==0.7.0 + # via pydantic +antlr4-python3-runtime==4.9.3 + # via omegaconf +appdirs==1.4.4 + # via cantilever +argklass==1.4.4 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/llm/requirements.in + # -r benchmarks/purejaxrl/requirements.in +astroid==3.3.8 + # via pylint +asttokens==3.0.0 + # via giving +async-timeout==5.0.1 + # via aiohttp +attrs==25.1.0 + # via + # aiohttp + # dm-tree +beartype==0.19.0 + # via -r benchmarks/vjepa/requirements.in +betterproto==2.0.0b6 + # via sigstore-protobuf-specs +black==25.1.0 + # via navix +blinker==1.9.0 + # via flask +blobfile==3.0.0 + # via + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt + # torchtune +blosc2==2.7.1 + # via tables +botorch==0.13.0 + # via -r benchmarks/recursiongfn/requirements.in +braceexpand==0.1.7 + # via + # -r benchmarks/vjepa/requirements.in + # webdataset +brax==0.12.1 + # via + # -r benchmarks/brax/requirements.in + # -r benchmarks/purejaxrl/requirements.in +cantilever==0.1.0 + # via -r benchmarks/torchatari/requirements.in +certifi==2025.1.31 + # via + # requests + # sentry-sdk +cffi==1.17.1 + # via cryptography +charset-normalizer==3.4.1 + # via requests +chex==0.1.88 + # via + # distrax + # evosax + # flashbax + # gymnax + # optax + # rlax +click==8.1.8 + # via + # black + # flask + # wandb +cloudpickle==3.1.1 + # via + # gym + # gymnasium + # submitit + # tensorflow-probability +codefind==0.1.7 + # via ptera +contourpy==1.3.1 + # via matplotlib +cryptography==44.0.0 + # via + # model-signing + # pyopenssl + # rfc3161-client + # sigstore +cvxopt==1.3.2 + # via -r benchmarks/recursiongfn/requirements.in +cycler==0.12.1 + # via matplotlib +datasets==3.2.0 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/llama/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/rlhf/requirements.in + # torchtune + # trl +decorator==5.1.1 + # via tensorflow-probability +decord==0.6.0 + # via -r benchmarks/vjepa/requirements.in +diffusers[torch]==0.32.2 + # via -r benchmarks/diffusion/requirements.in +dill==0.3.8 + # via + # datasets + # multiprocess + # pylint +distrax==0.1.5 + # via + # -r benchmarks/purejaxrl/requirements.in + # rlax +dm-env==1.6 + # via + # brax + # envpool + # rlax +dm-tree==0.1.9 + # via + # dm-env + # tensorflow-probability +dnspython==2.7.0 + # via email-validator +docker-pycreds==0.4.0 + # via wandb +docstring-parser==0.16 + # via tyro +dotmap==1.3.30 + # via evosax +einops==0.8.0 + # via + # -r benchmarks/rlhf/requirements.in + # -r benchmarks/vjepa/requirements.in +email-validator==2.2.0 + # via pydantic +envpool==0.8.4 + # via -r benchmarks/torchatari/requirements.in +etils[epath,epy]==1.11.0 + # via + # brax + # mujoco + # mujoco-mjx + # optax + # orbax-checkpoint +evosax==0.1.6 + # via -r benchmarks/purejaxrl/requirements.in +exceptiongroup==1.2.2 + # via pytest +executing==2.2.0 + # via varname +fairscale==0.4.13 + # via + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt + # -r benchmarks/llama/requirements.in + # -r benchmarks/llm/requirements.in +farama-notifications==0.0.4 + # via gymnasium +filelock==3.17.0 + # via + # blobfile + # datasets + # diffusers + # huggingface-hub + # torch + # transformers +fire==0.7.0 + # via + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt + # -r benchmarks/llama/requirements.in +flake8==7.1.1 + # via navix +flashbax==0.1.2 + # via -r benchmarks/purejaxrl/requirements.in +flask==3.1.0 + # via + # brax + # flask-cors +flask-cors==5.0.0 + # via brax +flax==0.10.2 + # via + # -r benchmarks/purejaxrl/requirements.in + # brax + # evosax + # flashbax + # gymnax + # navix +fonttools==4.55.8 + # via matplotlib +frozenlist==1.5.0 + # via + # aiohttp + # aiosignal +fsspec[http]==2024.9.0 + # via + # datasets + # etils + # huggingface-hub + # lightning + # pytorch-lightning + # torch + # torch-geometric +fvcore==0.1.5.post20221221 + # via -r benchmarks/dinov2/requirements.in +gast==0.6.0 + # via tensorflow-probability +gitdb==4.0.12 + # via gitpython +gitpython==3.1.44 + # via + # -r benchmarks/recursiongfn/requirements.in + # wandb +giving==0.4.3 + # via + # ptera + # voir +glfw==2.8.0 + # via mujoco +gpytorch==1.14 + # via + # -r benchmarks/recursiongfn/requirements.in + # botorch +grpcio==1.70.0 + # via + # brax + # tensorboard +grpclib==0.4.7 + # via betterproto +gym==0.26.2 + # via + # -r benchmarks/torchatari/requirements.in + # brax + # envpool + # gymnax +gym-notices==0.0.8 + # via gym +gymnasium==1.0.0 + # via + # envpool + # gymnax +gymnax==0.0.8 + # via + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt + # -r benchmarks/purejaxrl/requirements.in +h2==4.2.0 + # via grpclib +hf-transfer==0.1.9 + # via huggingface-hub +hjson==3.1.0 + # via argklass +hpack==4.1.0 + # via h2 +huggingface-hub[hf-transfer]==0.28.1 + # via + # -r benchmarks/timm/requirements.in + # accelerate + # datasets + # diffusers + # timm + # tokenizers + # torchtune + # transformers +humanize==4.11.0 + # via orbax-checkpoint +hyperframe==6.1.0 + # via h2 +id==1.5.0 + # via sigstore +idna==3.10 + # via + # email-validator + # requests + # yarl +importlib-metadata==8.6.1 + # via diffusers +importlib-resources==5.13.0 + # via + # argklass + # cantilever + # etils + # sigstore + # torchcompat +in-toto-attestation==0.9.3 + # via model-signing +iniconfig==2.0.0 + # via pytest +iopath==0.1.10 + # via + # -r benchmarks/dinov2/requirements.in + # fvcore +isort==6.0.0 + # via pylint +itsdangerous==2.2.0 + # via flask +jax==0.5.0 + # via + # -r benchmarks/brax/requirements.in + # -r benchmarks/purejaxrl/requirements.in + # brax + # chex + # distrax + # evosax + # flashbax + # flax + # gymnax + # jaxopt + # mujoco-mjx + # optax + # orbax-checkpoint + # rlax +jaxlib==0.5.0 + # via + # brax + # chex + # distrax + # evosax + # flashbax + # gymnax + # jax + # jaxopt + # mujoco-mjx + # optax + # rlax +jaxopt==0.8.3 + # via brax +jaxtyping==0.2.37 + # via + # gpytorch + # linear-operator +jinja2==3.1.5 + # via + # brax + # flask + # torch + # torch-geometric +joblib==1.4.2 + # via scikit-learn +kagglehub==0.3.7 + # via torchtune +kiwisolver==1.4.8 + # via matplotlib +lightning==2.5.0.post0 + # via -r benchmarks/lightning/requirements.in +lightning-utilities==0.12.0 + # via + # lightning + # pytorch-lightning + # torchmetrics +linear-operator==0.6 + # via + # botorch + # gpytorch +lxml==5.3.0 + # via blobfile +markdown==3.7 + # via tensorboard +markdown-it-py==3.0.0 + # via rich +markupsafe==3.0.2 + # via + # jinja2 + # werkzeug +matplotlib==3.10.0 + # via + # evosax + # gymnax + # seaborn +mccabe==0.7.0 + # via + # flake8 + # pylint +mdurl==0.1.2 + # via markdown-it-py +ml-collections==1.0.0 + # via brax +ml-dtypes==0.5.1 + # via + # jax + # jaxlib + # tensorstore +model-signing==0.2.0 + # via kagglehub +mpmath==1.3.0 + # via + # gpytorch + # linear-operator + # sympy +msgpack==1.1.0 + # via + # blosc2 + # flax + # orbax-checkpoint +mujoco==3.2.7 + # via + # brax + # mujoco-mjx +mujoco-mjx==3.2.7 + # via brax +multidict==6.1.0 + # via + # aiohttp + # grpclib + # yarl +multipledispatch==1.0.0 + # via botorch +multiprocess==0.70.16 + # via datasets +mypy-extensions==1.0.0 + # via + # black + # typing-inspect +navix==0.7.0 + # via -r benchmarks/purejaxrl/requirements.in +ndindex==1.9.2 + # via blosc2 +nest-asyncio==1.6.0 + # via orbax-checkpoint +networkx==3.4.2 + # via + # -r benchmarks/recursiongfn/requirements.in + # torch +numexpr==2.10.2 + # via + # blosc2 + # tables +numpy==1.26.4 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/purejaxrl/requirements.in + # -r benchmarks/torchatari/requirements.in + # -r benchmarks/vjepa/requirements.in + # accelerate + # blosc2 + # brax + # chex + # contourpy + # datasets + # decord + # diffusers + # distrax + # dm-env + # dm-tree + # envpool + # evosax + # fairscale + # flashbax + # fvcore + # gym + # gymnasium + # jax + # jaxlib + # jaxopt + # matplotlib + # ml-dtypes + # mujoco + # navix + # numexpr + # opencv-python + # optax + # orbax-checkpoint + # pandas + # pyro-ppl + # rdkit + # rlax + # scikit-learn + # scipy + # seaborn + # tables + # tensorboard + # tensorboardx + # tensorflow-probability + # tensorstore + # torch-geometric + # torchmetrics + # torchtune + # torchvision + # transformers + # trimesh + # webdataset + # xformers +nvidia-ml-py==12.570.86 + # via voir +omegaconf==2.3.0 + # via + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/recursiongfn/requirements.in + # torchtune + # voir +opencv-python==4.11.0.86 + # via -r benchmarks/vjepa/requirements.in +opt-einsum==3.4.0 + # via + # jax + # pyro-ppl +optax==0.2.4 + # via + # -r benchmarks/purejaxrl/requirements.in + # brax + # flax +optree==0.14.0 + # via envpool +orbax-checkpoint==0.11.1 + # via + # brax + # flax +ovld==0.3.9 + # via voir +packaging==24.2 + # via + # accelerate + # black + # datasets + # envpool + # huggingface-hub + # kagglehub + # lightning + # lightning-utilities + # matplotlib + # pytest + # pytorch-lightning + # setuptools-scm + # tables + # tensorboard + # tensorboardx + # torchmetrics + # transformers +pandas==2.2.3 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in + # -r benchmarks/vjepa/requirements.in + # datasets + # seaborn +pathspec==0.12.1 + # via black +pillow==11.1.0 + # via + # -r benchmarks/huggingface/requirements.in + # -r benchmarks/llava/requirements.in + # brax + # diffusers + # fvcore + # matplotlib + # navix + # rdkit + # torchtune + # torchvision +platformdirs==4.3.6 + # via + # black + # pylint + # sigstore + # wandb +pluggy==1.5.0 + # via pytest +portalocker==3.1.1 + # via iopath +propcache==0.2.1 + # via + # aiohttp + # yarl +protobuf==5.29.3 + # via + # in-toto-attestation + # orbax-checkpoint + # tensorboard + # tensorboardx + # wandb +psutil==5.9.8 + # via + # accelerate + # torch-geometric + # torchtune + # voir + # wandb +ptera==1.4.1 + # via voir +py-cpuinfo==9.0.0 + # via + # blosc2 + # tables +pyarrow==19.0.0 + # via + # -r benchmarks/recursiongfn/requirements.in + # datasets +pyasn1==0.6.1 + # via sigstore +pycodestyle==2.12.1 + # via flake8 +pycparser==2.22 + # via cffi +pycryptodomex==3.21.0 + # via blobfile +pydantic[email]==2.10.6 + # via + # sigstore + # sigstore-rekor-types + # wandb +pydantic-core==2.27.2 + # via pydantic +pyflakes==3.2.0 + # via flake8 +pygments==2.19.1 + # via rich +pyjwt==2.10.1 + # via sigstore +pylint==3.3.4 + # via navix +pyopengl==3.1.9 + # via mujoco +pyopenssl==25.0.0 + # via sigstore +pyparsing==3.2.1 + # via + # matplotlib + # torch-geometric +pyre-extensions==0.0.32 + # via botorch +pyro-api==0.1.2 + # via pyro-ppl +pyro-ppl==1.9.1 + # via + # -r benchmarks/recursiongfn/requirements.in + # botorch +pytest==8.3.4 + # via navix +python-dateutil==2.9.0.post0 + # via + # betterproto + # matplotlib + # pandas +pytinyrenderer==0.0.14 + # via brax +pytorch-lightning==2.5.0.post0 + # via lightning +pytorch-triton-rocm==3.2.0 + # via torch +pytz==2025.1 + # via pandas +pyyaml==6.0.2 + # via + # -r benchmarks/llm/requirements.in + # -r benchmarks/timm/requirements.in + # -r benchmarks/vjepa/requirements.in + # accelerate + # datasets + # evosax + # flax + # fvcore + # gymnax + # huggingface-hub + # lightning + # ml-collections + # omegaconf + # orbax-checkpoint + # pytorch-lightning + # timm + # transformers + # wandb + # webdataset + # yacs +rdkit==2024.9.4 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in +reactivex==4.0.4 + # via giving +regex==2024.11.6 + # via + # diffusers + # tiktoken + # transformers +requests==2.32.3 + # via + # datasets + # diffusers + # huggingface-hub + # id + # kagglehub + # sigstore + # tiktoken + # torch-geometric + # transformers + # tuf + # wandb +rfc3161-client==0.1.2 + # via sigstore +rfc8785==0.1.4 + # via sigstore +rich==13.9.4 + # via + # flax + # sigstore + # trl + # tyro + # voir +rlax==0.1.6 + # via navix +safetensors==0.5.2 + # via + # -r benchmarks/timm/requirements.in + # accelerate + # diffusers + # timm + # torchtune + # transformers +scikit-learn==1.6.1 + # via gpytorch +scipy==1.15.1 + # via + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/recursiongfn/requirements.in + # botorch + # brax + # gpytorch + # jax + # jaxlib + # jaxopt + # linear-operator + # mujoco-mjx + # scikit-learn + # torch-cluster + # torch-sparse +seaborn==0.13.2 + # via gymnax +securesystemslib==1.2.0 + # via tuf +sentencepiece==0.2.0 + # via + # -r benchmarks/llama/requirements.in + # torchtune +sentry-sdk==2.20.0 + # via wandb +setproctitle==1.3.4 + # via wandb +setuptools-scm==8.1.0 + # via navix +shtab==1.7.1 + # via tyro +sigstore==3.6.1 + # via model-signing +sigstore-protobuf-specs==0.3.2 + # via sigstore +sigstore-rekor-types==0.0.18 + # via sigstore +simplejson==3.19.3 + # via orbax-checkpoint +six==1.17.0 + # via + # docker-pycreds + # ml-collections + # python-dateutil + # tensorboard + # tensorflow-probability +smmap==5.0.2 + # via gitdb +submitit==1.5.2 + # via + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/vjepa/requirements.in +sympy==1.13.1 + # via torch +tables==3.10.1 + # via -r benchmarks/recursiongfn/requirements.in +tabulate==0.9.0 + # via fvcore +tensorboard==2.18.0 + # via + # -r benchmarks/recursiongfn/requirements.in + # -r benchmarks/torchatari/requirements.in +tensorboard-data-server==0.7.2 + # via tensorboard +tensorboardx==2.6.2.2 + # via brax +tensorflow-probability==0.25.0 + # via distrax +tensorstore==0.1.71 + # via + # flashbax + # flax + # orbax-checkpoint +termcolor==2.5.0 + # via + # fire + # fvcore +threadpoolctl==3.5.0 + # via + # botorch + # scikit-learn +tiktoken==0.8.0 + # via torchtune +timm==1.0.14 + # via -r benchmarks/vjepa/requirements.in +tokenizers==0.21.0 + # via transformers +tomli==2.2.1 + # via + # black + # pylint + # pytest + # setuptools-scm +tomlkit==0.13.2 + # via pylint +toolz==1.0.0 + # via chex +torch==2.6.0+rocm6.2.4 + # via + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt + # -r benchmarks/brax/requirements.in + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/flops/requirements.in + # -r benchmarks/geo_gnn/requirements-pre.in + # -r benchmarks/huggingface/requirements.in + # -r benchmarks/lightning/requirements.in + # -r benchmarks/llama/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/llm/requirements.in + # -r benchmarks/purejaxrl/requirements.in + # -r benchmarks/recursiongfn/requirements.in + # -r benchmarks/rlhf/requirements.in + # -r benchmarks/timm/requirements.in + # -r benchmarks/torchatari/requirements.in + # -r benchmarks/torchvision/requirements.in + # -r benchmarks/torchvision_ddp/requirements.in + # -r benchmarks/vjepa/requirements.in + # accelerate + # botorch + # diffusers + # fairscale + # lightning + # linear-operator + # pyro-ppl + # pytorch-lightning + # timm + # torchmetrics + # torchvision + # xformers +torch-cluster==1.6.3 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in +torch-geometric==2.6.1 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in +torch-scatter==2.1.2 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in +torch-sparse==0.6.18 + # via + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/recursiongfn/requirements.in +torchao==0.8.0+rocm6.2.4 + # via -r benchmarks/llm/requirements.in +torchcompat==1.1.4 + # via + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt + # -r benchmarks/flops/requirements.in + # -r benchmarks/lightning/requirements.in + # -r benchmarks/torchatari/requirements.in + # -r benchmarks/torchvision/requirements.in + # -r benchmarks/torchvision_ddp/requirements.in +torchmetrics==1.6.1 + # via + # -r benchmarks/dinov2/requirements.in + # lightning + # pytorch-lightning +torchtune==0.5.0 + # via -r benchmarks/llm/requirements.in +torchvision==0.21.0+rocm6.2.4 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/flops/requirements.in + # -r benchmarks/lightning/requirements.in + # -r benchmarks/timm/requirements.in + # -r benchmarks/torchvision/requirements.in + # -r benchmarks/torchvision_ddp/requirements.in + # -r benchmarks/vjepa/requirements.in + # timm +tqdm==4.67.1 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/flops/requirements.in + # -r benchmarks/torchvision/requirements.in + # -r benchmarks/torchvision_ddp/requirements.in + # datasets + # fvcore + # huggingface-hub + # iopath + # kagglehub + # lightning + # pyro-ppl + # pytorch-lightning + # torch-geometric + # torchtune + # transformers +transformers==4.48.2 + # via + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/huggingface/requirements.in + # -r benchmarks/llama/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/llm/requirements.in + # -r benchmarks/rlhf/requirements.in + # trl +trimesh==4.6.1 + # via + # brax + # mujoco-mjx +trl==0.14.0 + # via -r benchmarks/rlhf/requirements.in +tuf==5.1.0 + # via sigstore +typeguard==4.4.1 + # via tyro +types-protobuf==5.29.1.20241207 + # via envpool +typing-extensions==4.12.2 + # via + # astroid + # black + # botorch + # brax + # chex + # envpool + # etils + # flashbax + # flax + # gymnasium + # huggingface-hub + # iopath + # lightning + # lightning-utilities + # model-signing + # multidict + # navix + # optree + # orbax-checkpoint + # pydantic + # pydantic-core + # pyopenssl + # pyre-extensions + # pytorch-lightning + # reactivex + # rich + # submitit + # tables + # torch + # typeguard + # typing-inspect + # tyro + # wandb +typing-inspect==0.9.0 + # via pyre-extensions +tyro==0.9.13 + # via + # -r benchmarks/torchatari/requirements.in + # navix +tzdata==2025.1 + # via pandas +urllib3==2.3.0 + # via + # blobfile + # requests + # sentry-sdk +varname==0.14.0 + # via giving +voir==0.2.19 + # via + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt + # -r benchmarks/brax/requirements.in + # -r benchmarks/diffusion/requirements.in + # -r benchmarks/dinov2/requirements.in + # -r benchmarks/flops/requirements.in + # -r benchmarks/geo_gnn/requirements.in + # -r benchmarks/huggingface/requirements.in + # -r benchmarks/lightning/requirements.in + # -r benchmarks/llama/requirements.in + # -r benchmarks/llava/requirements.in + # -r benchmarks/llm/requirements.in + # -r benchmarks/purejaxrl/requirements.in + # -r benchmarks/recursiongfn/requirements.in + # -r benchmarks/rlhf/requirements.in + # -r benchmarks/timm/requirements.in + # -r benchmarks/torchatari/requirements.in + # -r benchmarks/torchvision/requirements.in + # -r benchmarks/torchvision_ddp/requirements.in + # -r benchmarks/vjepa/requirements.in +wadler-lindig==0.1.3 + # via jaxtyping +wandb==0.19.5 + # via + # -r benchmarks/recursiongfn/requirements.in + # navix +webdataset==0.2.100 + # via -r benchmarks/vjepa/requirements.in +werkzeug==3.1.3 + # via + # flask + # tensorboard +wrapt==1.17.2 + # via dm-tree +xformers==0.0.29.post2 + # via -r benchmarks/dinov2/requirements.in +xxhash==3.5.0 + # via datasets +yacs==0.1.8 + # via fvcore +yarl==1.18.3 + # via aiohttp +zipp==3.21.0 + # via + # etils + # importlib-metadata + +# The following packages are considered to be unsafe in a requirements file: +# setuptools diff --git a/benchmarks/brax/requirements.cuda.txt b/benchmarks/brax/requirements.cuda.txt index 8ced94d0..ee3b58d6 100644 --- a/benchmarks/brax/requirements.cuda.txt +++ b/benchmarks/brax/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/brax/requirements.cuda.txt .pin/tmp-constraints-cuda-brax.txt benchmarks/brax/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -16,6 +16,7 @@ absl-py==2.1.0 # brax # chex # dm-env + # dm-tree # ml-collections # mujoco # mujoco-mjx @@ -29,6 +30,10 @@ asttokens==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # giving +attrs==25.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree blinker==1.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -57,7 +62,7 @@ dm-env==1.6 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax -dm-tree==0.1.8 +dm-tree==0.1.9 # via # -c .pin/../.pin/constraints-cuda-torch.txt # dm-env @@ -69,7 +74,7 @@ etils[epath,epy]==1.11.0 # mujoco-mjx # optax # orbax-checkpoint -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -77,7 +82,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton flask==3.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -105,7 +109,7 @@ glfw==2.8.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # mujoco -grpcio==1.69.0 +grpcio==1.70.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax @@ -121,7 +125,7 @@ humanize==4.11.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # orbax-checkpoint -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # etils @@ -223,6 +227,7 @@ numpy==1.26.4 # brax # chex # dm-env + # dm-tree # gym # jax # jaxlib @@ -236,57 +241,61 @@ numpy==1.26.4 # tensorstore # trimesh # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -299,9 +308,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -318,7 +329,7 @@ optax==0.2.4 # -c .pin/../.pin/constraints-cuda-torch.txt # brax # flax -orbax-checkpoint==0.11.0 +orbax-checkpoint==0.11.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax @@ -409,18 +420,18 @@ toolz==1.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # chex -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/brax/requirements.in # xformers -trimesh==4.6.0 +trimesh==4.6.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax # mujoco-mjx -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -448,7 +459,11 @@ werkzeug==3.1.3 # via # -c .pin/../.pin/constraints-cuda-torch.txt # flask -xformers==0.0.29.post1 +wrapt==1.17.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/brax/requirements.rocm.txt b/benchmarks/brax/requirements.rocm.txt index e0a9eed6..0ce09ba8 100644 --- a/benchmarks/brax/requirements.rocm.txt +++ b/benchmarks/brax/requirements.rocm.txt @@ -5,15 +5,16 @@ # pip-compile --output-file=benchmarks/brax/requirements.rocm.txt .pin/tmp-constraints-rocm-brax.txt benchmarks/brax/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com absl-py==2.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # dm-env + # dm-tree # ml-collections # mujoco # mujoco-mjx @@ -21,115 +22,118 @@ absl-py==2.1.0 # orbax-checkpoint antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving +attrs==25.1.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree blinker==1.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask brax==0.12.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/brax/requirements.in chex==0.1.88 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # optax click==8.1.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask cloudpickle==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera dm-env==1.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax -dm-tree==0.1.8 +dm-tree==0.1.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # dm-env etils[epath,epy]==1.11.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco # mujoco-mjx # optax # orbax-checkpoint -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch flask==3.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flask-cors flask-cors==5.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax flax==0.10.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # etils # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir glfw==2.8.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # mujoco -grpcio==1.69.0 +grpcio==1.70.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax gym==0.26.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax gym-notices==0.0.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym humanize==4.11.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # etils itsdangerous==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask -jax==0.4.38 +jax==0.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/brax/requirements.in # brax # chex @@ -138,9 +142,9 @@ jax==0.4.38 # mujoco-mjx # optax # orbax-checkpoint -jaxlib==0.4.38 +jaxlib==0.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # jax @@ -149,69 +153,70 @@ jaxlib==0.4.38 # optax jaxopt==0.8.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flask # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 # werkzeug mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py ml-collections==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax ml-dtypes==0.5.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jax # jaxlib # tensorstore mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy msgpack==1.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # orbax-checkpoint mujoco==3.2.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco-mjx mujoco-mjx==3.2.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax nest-asyncio==1.6.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # dm-env + # dm-tree # gym # jax # jaxlib @@ -224,88 +229,88 @@ numpy==1.26.4 # tensorboardx # tensorstore # trimesh -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir opt-einsum==3.4.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jax optax==0.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flax -orbax-checkpoint==0.11.0 +orbax-checkpoint==0.11.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flax ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboardx pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax protobuf==5.29.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint # tensorboardx psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pyopengl==3.1.7 +pyopengl==3.1.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # mujoco pytinyrenderer==0.0.14 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # ml-collections # omegaconf # orbax-checkpoint reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # voir scipy==1.15.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # jax # jaxlib @@ -313,41 +318,41 @@ scipy==1.15.1 # mujoco-mjx simplejson==3.19.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ml-collections sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tensorboardx==2.6.2.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax tensorstore==0.1.71 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # orbax-checkpoint toolz==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # chex -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/brax/requirements.in -trimesh==4.5.3 +trimesh==4.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco-mjx typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # etils @@ -358,18 +363,22 @@ typing-extensions==4.12.2 # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/brax/requirements.in werkzeug==3.1.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask +wrapt==1.17.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree zipp==3.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # etils diff --git a/benchmarks/diffusion/requirements.cuda.txt b/benchmarks/diffusion/requirements.cuda.txt index a84caa37..f4f69b2f 100644 --- a/benchmarks/diffusion/requirements.cuda.txt +++ b/benchmarks/diffusion/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/diffusion/requirements.cuda.txt .pin/tmp-constraints-cuda-diffusion-nodes.txt benchmarks/diffusion/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -44,11 +44,11 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -73,7 +73,7 @@ dill==0.3.8 # -c .pin/../.pin/constraints-cuda-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -85,7 +85,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -106,7 +105,7 @@ hjson==3.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # argklass -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # accelerate @@ -123,7 +122,7 @@ importlib-metadata==8.6.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # diffusers -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # argklass @@ -195,57 +194,61 @@ numpy==1.26.4 # torchvision # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -258,9 +261,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -318,7 +323,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -373,7 +378,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -381,7 +386,7 @@ torch==2.5.1+cu121 # diffusers # torchvision # xformers -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/diffusion/requirements.in @@ -392,11 +397,11 @@ tqdm==4.67.1 # datasets # huggingface-hub # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/diffusion/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -425,7 +430,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/diffusion/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/diffusion/requirements.rocm.txt b/benchmarks/diffusion/requirements.rocm.txt index 4a68dac9..b7255398 100644 --- a/benchmarks/diffusion/requirements.rocm.txt +++ b/benchmarks/diffusion/requirements.rocm.txt @@ -5,108 +5,107 @@ # pip-compile --output-file=benchmarks/diffusion/requirements.rocm.txt .pin/tmp-constraints-rocm-diffusion-nodes.txt benchmarks/diffusion/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com -accelerate==1.2.1 +accelerate==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in # diffusers aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf argklass==1.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera datasets==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in diffusers[torch]==0.32.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # diffusers # huggingface-hub - # pytorch-triton-rocm # torch # transformers frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir hjson==3.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # diffusers @@ -114,124 +113,124 @@ huggingface-hub==0.27.1 # transformers idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl -importlib-metadata==8.5.0 +importlib-metadata==8.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # diffusers -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl multiprocess==0.70.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # diffusers # pandas # torchvision # transformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub # transformers pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # diffusers # torchvision propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub @@ -239,97 +238,97 @@ pyyaml==6.0.2 # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # diffusers # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # diffusers # huggingface-hub # transformers rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # diffusers # transformers six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # diffusers # torchvision -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in # datasets # huggingface-hub # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/diffusion/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # multidict # reactivex # rich # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/diffusion/requirements.in xxhash==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp zipp==3.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # importlib-metadata diff --git a/benchmarks/dinov2/requirements.cuda.txt b/benchmarks/dinov2/requirements.cuda.txt index 42f002e6..fd4eb9d5 100644 --- a/benchmarks/dinov2/requirements.cuda.txt +++ b/benchmarks/dinov2/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/dinov2/requirements.cuda.txt .pin/tmp-constraints-cuda-dinov2-giant-gpus.txt benchmarks/dinov2/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -26,7 +26,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -34,7 +34,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -73,7 +72,7 @@ jinja2==3.1.5 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -lightning-utilities==0.11.9 +lightning-utilities==0.12.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchmetrics @@ -113,57 +112,61 @@ numpy==1.26.4 # torchmetrics # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -176,9 +179,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -257,7 +262,7 @@ termcolor==2.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # fvcore -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -269,7 +274,7 @@ torchmetrics==1.6.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/dinov2/requirements.in -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/dinov2/requirements.in @@ -278,7 +283,7 @@ tqdm==4.67.1 # -c .pin/../.pin/constraints-cuda-torch.txt # fvcore # iopath -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -300,7 +305,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/dinov2/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/dinov2/requirements.rocm.txt b/benchmarks/dinov2/requirements.rocm.txt index ece66eaf..0087b46a 100644 --- a/benchmarks/dinov2/requirements.rocm.txt +++ b/benchmarks/dinov2/requirements.rocm.txt @@ -5,188 +5,187 @@ # pip-compile --output-file=benchmarks/dinov2/requirements.rocm.txt .pin/tmp-constraints-rocm-dinov2-giant-gpus.txt benchmarks/dinov2/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving cloudpickle==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # submitit codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fvcore==0.1.5.post20221221 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir iopath==0.1.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in # fvcore jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -lightning-utilities==0.11.9 +lightning-utilities==0.12.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchmetrics markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore # scipy # torchmetrics # torchvision # xformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning-utilities # torchmetrics pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore # torchvision portalocker==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # iopath psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore # omegaconf # yacs reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir scipy==1.15.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in submitit==1.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tabulate==0.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore termcolor==2.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in # torchmetrics # torchvision # xformers torchmetrics==1.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore # iopath typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # iopath # lightning-utilities # reactivex @@ -195,20 +194,20 @@ typing-extensions==4.12.2 # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/dinov2/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/dinov2/requirements.in yacs==0.1.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fvcore # The following packages are considered to be unsafe in a requirements file: diff --git a/benchmarks/flops/requirements.cuda.txt b/benchmarks/flops/requirements.cuda.txt index dc9bd142..d229b927 100644 --- a/benchmarks/flops/requirements.cuda.txt +++ b/benchmarks/flops/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/flops/requirements.cuda.txt .pin/tmp-constraints-cuda-flops.txt benchmarks/flops/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -22,7 +22,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -30,7 +30,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -40,7 +39,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchcompat @@ -98,57 +97,61 @@ numpy==1.26.4 # scipy # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -161,9 +164,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -216,7 +221,7 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -228,7 +233,7 @@ torchcompat==1.1.4 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/flops/requirements.in -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/flops/requirements.in @@ -236,7 +241,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/flops/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -255,7 +260,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/flops/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/flops/requirements.rocm.txt b/benchmarks/flops/requirements.rocm.txt index 8b310c1b..192c1445 100644 --- a/benchmarks/flops/requirements.rocm.txt +++ b/benchmarks/flops/requirements.rocm.txt @@ -5,149 +5,148 @@ # pip-compile --output-file=benchmarks/flops/requirements.rocm.txt .pin/tmp-constraints-rocm-flops.txt benchmarks/flops/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchcompat jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/flops/requirements.in # torchvision torchcompat==1.1.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/flops/requirements.in -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/flops/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/flops/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # reactivex # rich # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/flops/requirements.in diff --git a/benchmarks/geo_gnn/requirements-pre.cuda.txt b/benchmarks/geo_gnn/requirements-pre.cuda.txt index df1909f1..d4109443 100644 --- a/benchmarks/geo_gnn/requirements-pre.cuda.txt +++ b/benchmarks/geo_gnn/requirements-pre.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/geo_gnn/requirements-pre.cuda.txt .pin/tmp-constraints-cuda-dimenet.txt benchmarks/geo_gnn/requirements-pre.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -14,7 +14,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -64,56 +63,60 @@ numpy==1.26.4 # ml-dtypes # scipy # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch nvidia-nccl-cu12==2.21.5 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -123,9 +126,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -142,13 +147,13 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/geo_gnn/requirements-pre.in # xformers -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -156,7 +161,7 @@ typing-extensions==4.12.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/geo_gnn/requirements-pre.rocm.txt b/benchmarks/geo_gnn/requirements-pre.rocm.txt index 32085fd4..9e90d4d5 100644 --- a/benchmarks/geo_gnn/requirements-pre.rocm.txt +++ b/benchmarks/geo_gnn/requirements-pre.rocm.txt @@ -5,47 +5,46 @@ # pip-compile --output-file=benchmarks/geo_gnn/requirements-pre.rocm.txt .pin/tmp-constraints-rocm-dimenet.txt benchmarks/geo_gnn/requirements-pre.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch diff --git a/benchmarks/geo_gnn/requirements.cuda.txt b/benchmarks/geo_gnn/requirements.cuda.txt index 9cd54346..1b91593c 100644 --- a/benchmarks/geo_gnn/requirements.cuda.txt +++ b/benchmarks/geo_gnn/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/geo_gnn/requirements.cuda.txt .pin/tmp-constraints-cuda-dimenet.txt benchmarks/geo_gnn/requirements-pre.cuda.txt benchmarks/geo_gnn/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -34,11 +34,11 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -50,7 +50,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -59,7 +59,6 @@ filelock==3.17.0 # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -154,59 +153,64 @@ numpy==1.26.4 # scipy # torch-geometric # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # -r benchmarks/geo_gnn/requirements-pre.cuda.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -219,9 +223,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt @@ -273,7 +279,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -314,7 +320,7 @@ sympy==1.13.1 # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -340,7 +346,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch-geometric -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/geo_gnn/requirements-pre.cuda.txt @@ -370,7 +376,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/geo_gnn/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/geo_gnn/requirements.rocm.txt b/benchmarks/geo_gnn/requirements.rocm.txt index 2a33b7f9..84959e35 100644 --- a/benchmarks/geo_gnn/requirements.rocm.txt +++ b/benchmarks/geo_gnn/requirements.rocm.txt @@ -5,261 +5,260 @@ # pip-compile --output-file=benchmarks/geo_gnn/requirements.rocm.txt .pin/tmp-constraints-rocm-dimenet.txt benchmarks/geo_gnn/requirements-pre.rocm.txt benchmarks/geo_gnn/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt - # pytorch-triton-rocm # torch frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # torch # torch-geometric giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # torch # torch-geometric markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in # pandas # rdkit # scipy # torch-geometric -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rdkit propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich pyparsing==3.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf rdkit==2024.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir scipy==1.15.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-cluster # torch-sparse six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt torch-cluster==1.6.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in torch-geometric==2.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in torch-scatter==2.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in torch-sparse==0.6.18 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/geo_gnn/requirements-pre.rocm.txt # multidict # reactivex # rich # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/geo_gnn/requirements.in yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp diff --git a/benchmarks/huggingface/requirements.cuda.txt b/benchmarks/huggingface/requirements.cuda.txt index ad775694..07dd570b 100644 --- a/benchmarks/huggingface/requirements.cuda.txt +++ b/benchmarks/huggingface/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/huggingface/requirements.cuda.txt .pin/tmp-constraints-cuda-hf.txt benchmarks/huggingface/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -18,7 +18,7 @@ asttokens==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # giving -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -30,7 +30,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -40,7 +40,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -51,7 +50,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tokenizers @@ -114,57 +113,61 @@ numpy==1.26.4 # scipy # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -177,9 +180,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -256,7 +261,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -267,11 +272,11 @@ tqdm==4.67.1 # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/huggingface/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -295,7 +300,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/huggingface/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/huggingface/requirements.rocm.txt b/benchmarks/huggingface/requirements.rocm.txt index 1df97f95..1bdcb329 100644 --- a/benchmarks/huggingface/requirements.rocm.txt +++ b/benchmarks/huggingface/requirements.rocm.txt @@ -5,189 +5,188 @@ # pip-compile --output-file=benchmarks/huggingface/requirements.rocm.txt .pin/tmp-constraints-rocm-hf.txt benchmarks/huggingface/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub - # pytorch-triton-rocm # torch # transformers fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tokenizers # transformers idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # transformers pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/huggingface/requirements.in psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # omegaconf # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # transformers rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/huggingface/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/huggingface/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # reactivex # rich # torch urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/huggingface/requirements.in diff --git a/benchmarks/lightning/requirements.cuda.txt b/benchmarks/lightning/requirements.cuda.txt index 08511440..19d287f8 100644 --- a/benchmarks/lightning/requirements.cuda.txt +++ b/benchmarks/lightning/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/lightning/requirements.cuda.txt .pin/tmp-constraints-cuda-lightning-gpus.txt benchmarks/lightning/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -34,7 +34,7 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp @@ -42,7 +42,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -50,7 +50,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -71,7 +70,7 @@ idna==3.10 # via # -c .pin/../.pin/constraints-cuda-torch.txt # yarl -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchcompat @@ -99,7 +98,7 @@ lightning==2.5.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/lightning/requirements.in -lightning-utilities==0.11.9 +lightning-utilities==0.12.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # lightning @@ -145,57 +144,61 @@ numpy==1.26.4 # torchmetrics # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -208,9 +211,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -281,7 +286,7 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -301,7 +306,7 @@ torchmetrics==1.6.1 # -c .pin/../.pin/constraints-cuda-torch.txt # lightning # pytorch-lightning -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/lightning/requirements.in @@ -310,7 +315,7 @@ tqdm==4.67.1 # -c .pin/../.pin/constraints-cuda-torch.txt # lightning # pytorch-lightning -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -333,7 +338,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/lightning/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/lightning/requirements.rocm.txt b/benchmarks/lightning/requirements.rocm.txt index b8e49c55..31de53cb 100644 --- a/benchmarks/lightning/requirements.rocm.txt +++ b/benchmarks/lightning/requirements.rocm.txt @@ -5,187 +5,186 @@ # pip-compile --output-file=benchmarks/lightning/requirements.rocm.txt .pin/tmp-constraints-rocm-lightning-gpus.txt benchmarks/lightning/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # pytorch-lightning # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # yarl -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchcompat jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch lightning==2.5.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/lightning/requirements.in -lightning-utilities==0.11.9 +lightning-utilities==0.12.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # pytorch-lightning # torchmetrics markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchmetrics # torchvision -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # lightning-utilities # pytorch-lightning # torchmetrics pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich pytorch-lightning==2.5.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # omegaconf # pytorch-lightning reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/lightning/requirements.in # lightning # pytorch-lightning @@ -193,26 +192,26 @@ torch==2.5.1+rocm6.2 # torchvision torchcompat==1.1.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/lightning/requirements.in torchmetrics==1.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # pytorch-lightning -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/lightning/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # pytorch-lightning typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # lightning # lightning-utilities # multidict @@ -222,16 +221,16 @@ typing-extensions==4.12.2 # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/lightning/requirements.in yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # The following packages are considered to be unsafe in a requirements file: diff --git a/benchmarks/llama/requirements.cuda.txt b/benchmarks/llama/requirements.cuda.txt index 8cd9db85..c84b2675 100644 --- a/benchmarks/llama/requirements.cuda.txt +++ b/benchmarks/llama/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/llama/requirements.cuda.txt .pin/tmp-constraints-cuda-llm.txt benchmarks/llama/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -35,11 +35,11 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -60,7 +60,7 @@ dill==0.3.8 # -c .pin/../.pin/constraints-cuda-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -75,7 +75,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton fire==0.7.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -96,7 +95,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # datasets @@ -173,57 +172,61 @@ numpy==1.26.4 # scipy # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -236,9 +239,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -289,7 +294,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -347,7 +352,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -360,11 +365,11 @@ tqdm==4.67.1 # datasets # huggingface-hub # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/llama/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -393,7 +398,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/llama/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/llama/requirements.rocm.txt b/benchmarks/llama/requirements.rocm.txt index 36d1b558..6109c820 100644 --- a/benchmarks/llama/requirements.rocm.txt +++ b/benchmarks/llama/requirements.rocm.txt @@ -5,295 +5,294 @@ # pip-compile --output-file=benchmarks/llama/requirements.rocm.txt .pin/tmp-constraints-rocm-llm.txt benchmarks/llama/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera datasets==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname fairscale==0.4.13 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub - # pytorch-triton-rocm # torch # transformers fire==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # tokenizers # transformers idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl multiprocess==0.70.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fairscale # pandas # transformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # omegaconf # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers sentencepiece==0.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch termcolor==2.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fire tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in # fairscale tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llama/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # multidict # reactivex # rich # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/llama/requirements.in xxhash==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp diff --git a/benchmarks/llava/requirements.cuda.txt b/benchmarks/llava/requirements.cuda.txt index a437d671..e79b392e 100644 --- a/benchmarks/llava/requirements.cuda.txt +++ b/benchmarks/llava/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/llava/requirements.cuda.txt .pin/tmp-constraints-cuda-llava-single.txt benchmarks/llava/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -39,11 +39,11 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -64,7 +64,7 @@ dill==0.3.8 # -c .pin/../.pin/constraints-cuda-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -75,7 +75,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -92,7 +91,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # accelerate @@ -171,57 +170,61 @@ numpy==1.26.4 # scipy # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -234,9 +237,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -293,7 +298,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -345,7 +350,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -358,11 +363,11 @@ tqdm==4.67.1 # datasets # huggingface-hub # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/llava/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -391,7 +396,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/llava/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/llava/requirements.rocm.txt b/benchmarks/llava/requirements.rocm.txt index 54515f81..279dd74d 100644 --- a/benchmarks/llava/requirements.rocm.txt +++ b/benchmarks/llava/requirements.rocm.txt @@ -5,208 +5,207 @@ # pip-compile --output-file=benchmarks/llava/requirements.rocm.txt .pin/tmp-constraints-rocm-llava-single.txt benchmarks/llava/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com -accelerate==1.2.1 +accelerate==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera datasets==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # multiprocess -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub - # pytorch-triton-rocm # torch # transformers frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # tokenizers # transformers idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl multiprocess==0.70.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in # accelerate # datasets # pandas # transformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub # transformers pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub @@ -214,84 +213,84 @@ pyyaml==6.0.2 # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # transformers six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in # accelerate tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llava/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # multidict # reactivex # rich # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/llava/requirements.in xxhash==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp diff --git a/benchmarks/llm/requirements.cuda.txt b/benchmarks/llm/requirements.cuda.txt index 5bf56ae8..9ba97d06 100644 --- a/benchmarks/llm/requirements.cuda.txt +++ b/benchmarks/llm/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/llm/requirements.cuda.txt .pin/tmp-constraints-cuda-llm-full-mp-nodes.txt benchmarks/llm/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -27,6 +27,10 @@ aiosignal==1.3.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp +annotated-types==0.7.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # pydantic antlr4-python3-runtime==4.9.3 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -43,19 +47,27 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp +betterproto==2.0.0b6 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore-protobuf-specs blobfile==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/llm/requirements.txt # torchtune -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests +cffi==1.17.1 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # cryptography charset-normalizer==3.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -64,6 +76,13 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera +cryptography==44.0.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # model-signing + # pyopenssl + # rfc3161-client + # sigstore datasets==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -73,7 +92,15 @@ dill==0.3.8 # -c .pin/../.pin/constraints-cuda-torch.txt # datasets # multiprocess -executing==2.1.0 +dnspython==2.7.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # email-validator +email-validator==2.2.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # pydantic +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -90,7 +117,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton fire==0.7.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -111,11 +137,23 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir +grpclib==0.4.7 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # betterproto +h2==4.2.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # grpclib hjson==3.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # argklass -huggingface-hub[hf-transfer,hf_transfer]==0.27.1 +hpack==4.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # h2 +huggingface-hub[hf-transfer,hf_transfer]==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # accelerate @@ -123,15 +161,29 @@ huggingface-hub[hf-transfer,hf_transfer]==0.27.1 # tokenizers # torchtune # transformers +hyperframe==6.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # h2 +id==1.5.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore idna==3.10 # via # -c .pin/../.pin/constraints-cuda-torch.txt + # email-validator # requests # yarl -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # argklass + # sigstore +in-toto-attestation==0.9.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # model-signing jax[cuda12]==0.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -152,7 +204,7 @@ jinja2==3.1.5 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -kagglehub==0.3.6 +kagglehub==0.3.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchtune @@ -177,6 +229,10 @@ ml-dtypes==0.5.1 # -c .pin/../.pin/constraints-cuda-torch.txt # jax # jaxlib +model-signing==0.2.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # kagglehub mpmath==1.3.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -185,6 +241,7 @@ multidict==6.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp + # grpclib # yarl multiprocess==0.70.16 # via @@ -208,57 +265,61 @@ numpy==1.26.4 # torchtune # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -271,9 +332,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -306,11 +369,19 @@ pillow==11.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchtune +platformdirs==4.3.6 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore propcache==0.2.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp # yarl +protobuf==5.29.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # in-toto-attestation psutil==5.9.8 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -325,19 +396,45 @@ pyarrow==19.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # datasets +pyasn1==0.6.1 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore +pycparser==2.22 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # cffi pycryptodomex==3.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # blobfile +pydantic[email]==2.10.6 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore + # sigstore-rekor-types +pydantic-core==2.27.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # pydantic pygments==2.19.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # rich +pyjwt==2.10.1 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore +pyopenssl==25.0.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt + # betterproto # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -364,12 +461,24 @@ requests==2.32.3 # -c .pin/../.pin/constraints-cuda-torch.txt # datasets # huggingface-hub + # id # kagglehub + # sigstore # tiktoken # transformers + # tuf +rfc3161-client==0.1.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore +rfc8785==0.1.4 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore rich==13.9.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore # voir safetensors==0.5.2 # via @@ -382,10 +491,26 @@ scipy==1.15.1 # -c .pin/../.pin/constraints-cuda-torch.txt # jax # jaxlib +securesystemslib==1.2.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # tuf sentencepiece==0.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchtune +sigstore==3.6.1 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # model-signing +sigstore-protobuf-specs==0.3.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore +sigstore-rekor-types==0.0.18 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore six==1.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -406,7 +531,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -415,7 +540,7 @@ torch==2.5.1+cu121 # accelerate # fairscale # xformers -torchao==0.8.0 +torchao==0.8.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/llm/requirements.in @@ -431,19 +556,27 @@ tqdm==4.67.1 # kagglehub # torchtune # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/llm/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch +tuf==5.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # sigstore typing-extensions==4.12.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub + # model-signing # multidict + # pydantic + # pydantic-core + # pyopenssl # reactivex # rich # torch @@ -465,7 +598,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/llm/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/llm/requirements.rocm.txt b/benchmarks/llm/requirements.rocm.txt index 59e657c8..d7d903a3 100644 --- a/benchmarks/llm/requirements.rocm.txt +++ b/benchmarks/llm/requirements.rocm.txt @@ -5,197 +5,258 @@ # pip-compile --output-file=benchmarks/llm/requirements.rocm.txt .pin/tmp-constraints-rocm-llm-full-mp-nodes.txt benchmarks/llm/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com -accelerate==1.2.1 +accelerate==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp +annotated-types==0.7.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # pydantic antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf argklass==1.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp +betterproto==2.0.0b6 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore-protobuf-specs blobfile==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -r benchmarks/llm/requirements.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt # torchtune -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests +cffi==1.17.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # cryptography charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera +cryptography==44.0.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # model-signing + # pyopenssl + # rfc3161-client + # sigstore datasets==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # multiprocess -executing==2.1.0 +dnspython==2.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # email-validator +email-validator==2.2.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # pydantic +executing==2.2.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname fairscale==0.4.13 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt # -r benchmarks/llm/requirements.in - # -r benchmarks/llm/requirements.txt -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blobfile # datasets # huggingface-hub - # pytorch-triton-rocm # torch # transformers fire==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -r benchmarks/llm/requirements.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir +grpclib==0.4.7 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # betterproto +h2==4.2.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # grpclib +hf-transfer==0.1.9 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # huggingface-hub hjson==3.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass -huggingface-hub[hf-transfer,hf_transfer]==0.27.1 +hpack==4.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # h2 +huggingface-hub[hf-transfer,hf_transfer]==0.28.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # tokenizers # torchtune # transformers +hyperframe==6.1.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # h2 +id==1.5.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # email-validator # requests # yarl -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass + # sigstore +in-toto-attestation==0.9.3 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # model-signing jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -kagglehub==0.3.6 +kagglehub==0.3.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune lxml==5.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blobfile markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py +model-signing==0.2.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # kagglehub mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp + # grpclib # yarl multiprocess==0.70.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # fairscale # pandas # torchtune # transformers -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub @@ -203,54 +264,88 @@ packaging==24.2 # transformers pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune +platformdirs==4.3.6 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl +protobuf==5.29.3 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # in-toto-attestation psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # torchtune # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets +pyasn1==0.6.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore +pycparser==2.22 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # cffi pycryptodomex==3.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blobfile +pydantic[email]==2.10.6 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore + # sigstore-rekor-types +pydantic-core==2.27.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # pydantic pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich +pyjwt==2.10.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore +pyopenssl==25.0.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # betterproto # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in # accelerate # datasets @@ -259,113 +354,149 @@ pyyaml==6.0.2 # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tiktoken # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub + # id # kagglehub + # sigstore # tiktoken # transformers + # tuf +rfc3161-client==0.1.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore +rfc8785==0.1.4 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # torchtune # transformers +securesystemslib==1.2.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # tuf sentencepiece==0.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune +sigstore==3.6.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # model-signing +sigstore-protobuf-specs==0.3.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore +sigstore-rekor-types==0.0.18 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch termcolor==2.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # fire tiktoken==0.8.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchtune tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -r /network/scratch/d/delaunap/shared/milabench/benchmarks/llm/requirements.txt # -r benchmarks/llm/requirements.in - # -r benchmarks/llm/requirements.txt # accelerate # fairscale -torchao==0.8.0 +torchao==0.8.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in torchtune==0.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # kagglehub # torchtune # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/llm/requirements.in +tuf==5.1.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # sigstore typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub + # model-signing # multidict + # pydantic + # pydantic-core + # pyopenssl # reactivex # rich # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blobfile # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/llm/requirements.in xxhash==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp diff --git a/benchmarks/purejaxrl/requirements.cuda.txt b/benchmarks/purejaxrl/requirements.cuda.txt index 416807f4..30aca47f 100644 --- a/benchmarks/purejaxrl/requirements.cuda.txt +++ b/benchmarks/purejaxrl/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/purejaxrl/requirements.cuda.txt .pin/tmp-constraints-cuda-ppo.txt benchmarks/purejaxrl/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -17,6 +17,7 @@ absl-py==2.1.0 # chex # distrax # dm-env + # dm-tree # ml-collections # mujoco # mujoco-mjx @@ -44,7 +45,11 @@ asttokens==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # giving -black==24.10.0 +attrs==25.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree +black==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # navix @@ -56,7 +61,7 @@ brax==0.12.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/purejaxrl/requirements.in -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -116,7 +121,7 @@ dm-env==1.6 # -c .pin/../.pin/constraints-cuda-torch.txt # brax # rlax -dm-tree==0.1.8 +dm-tree==0.1.9 # via # -c .pin/../.pin/constraints-cuda-torch.txt # dm-env @@ -149,7 +154,7 @@ exceptiongroup==1.2.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pytest -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -161,7 +166,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton flake8==7.1.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -188,7 +192,7 @@ flax==0.10.2 # flashbax # gymnax # navix -fonttools==4.55.4 +fonttools==4.55.8 # via # -c .pin/../.pin/constraints-cuda-torch.txt # matplotlib @@ -218,7 +222,7 @@ glfw==2.8.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # mujoco -grpcio==1.69.0 +grpcio==1.70.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax @@ -252,7 +256,7 @@ idna==3.10 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # argklass @@ -261,7 +265,7 @@ iniconfig==2.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pytest -isort==5.13.2 +isort==6.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pylint @@ -399,6 +403,7 @@ numpy==1.26.4 # contourpy # distrax # dm-env + # dm-tree # evosax # flashbax # gym @@ -421,57 +426,61 @@ numpy==1.26.4 # tensorstore # trimesh # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -484,9 +493,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -504,7 +515,7 @@ optax==0.2.4 # -r benchmarks/purejaxrl/requirements.in # brax # flax -orbax-checkpoint==0.11.0 +orbax-checkpoint==0.11.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax @@ -564,7 +575,7 @@ pycodestyle==2.12.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # flake8 -pydantic==2.10.5 +pydantic==2.10.6 # via # -c .pin/../.pin/constraints-cuda-torch.txt # wandb @@ -580,7 +591,7 @@ pygments==2.19.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # rich -pylint==3.3.3 +pylint==3.3.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # navix @@ -605,7 +616,7 @@ pytinyrenderer==0.0.14 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -713,18 +724,18 @@ toolz==1.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # chex -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/purejaxrl/requirements.in # xformers -trimesh==4.6.0 +trimesh==4.6.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # brax # mujoco-mjx -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -753,7 +764,7 @@ typing-extensions==4.12.2 # typeguard # tyro # wandb -tyro==0.9.12 +tyro==0.9.13 # via # -c .pin/../.pin/constraints-cuda-torch.txt # navix @@ -775,7 +786,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/purejaxrl/requirements.in -wandb==0.19.4 +wandb==0.19.5 # via # -c .pin/../.pin/constraints-cuda-torch.txt # navix @@ -783,7 +794,11 @@ werkzeug==3.1.3 # via # -c .pin/../.pin/constraints-cuda-torch.txt # flask -xformers==0.0.29.post1 +wrapt==1.17.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/purejaxrl/requirements.rocm.txt b/benchmarks/purejaxrl/requirements.rocm.txt index b75c285d..f7be2e26 100644 --- a/benchmarks/purejaxrl/requirements.rocm.txt +++ b/benchmarks/purejaxrl/requirements.rocm.txt @@ -5,16 +5,17 @@ # pip-compile --output-file=benchmarks/purejaxrl/requirements.rocm.txt .pin/tmp-constraints-rocm-ppo.txt benchmarks/purejaxrl/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com absl-py==2.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # distrax # dm-env + # dm-tree # ml-collections # mujoco # mujoco-mjx @@ -24,48 +25,52 @@ absl-py==2.1.0 # tensorflow-probability annotated-types==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pydantic antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf argklass==1.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in astroid==3.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pylint asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving -black==24.10.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree +black==25.1.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix blinker==1.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask brax==0.12.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # sentry-sdk charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests chex==0.1.88 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # distrax # evosax # flashbax @@ -74,66 +79,66 @@ chex==0.1.88 # rlax click==8.1.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black # flask # wandb cloudpickle==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym # gymnasium # tensorflow-probability codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera contourpy==1.3.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib cycler==0.12.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib decorator==5.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorflow-probability dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pylint distrax==0.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in # rlax dm-env==1.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # rlax -dm-tree==0.1.8 +dm-tree==0.1.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # dm-env # tensorflow-probability docker-pycreds==0.4.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb docstring-parser==0.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro dotmap==1.3.30 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # evosax etils[epath,epy]==1.11.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco # mujoco-mjx @@ -141,135 +146,134 @@ etils[epath,epy]==1.11.0 # orbax-checkpoint evosax==0.1.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in exceptiongroup==1.2.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pytest -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname farama-notifications==0.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gymnasium -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch flake8==7.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix flashbax==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in flask==3.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flask-cors flask-cors==5.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax flax==0.10.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in # brax # evosax # flashbax # gymnax # navix -fonttools==4.55.3 +fonttools==4.55.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # etils # torch gast==0.6.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorflow-probability gitdb==4.0.12 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gitpython gitpython==3.1.44 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir glfw==2.8.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # mujoco -grpcio==1.69.0 +grpcio==1.70.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax gym==0.26.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # gymnax gym-notices==0.0.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym gymnasium==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gymnax gymnax==0.0.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/purejaxrl/requirements.in hjson==3.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass humanize==4.11.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # argklass # etils iniconfig==2.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pytest -isort==5.13.2 +isort==6.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pylint itsdangerous==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask -jax==0.4.38 +jax==0.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in # brax # chex @@ -283,9 +287,9 @@ jax==0.4.38 # optax # orbax-checkpoint # rlax -jaxlib==0.4.38 +jaxlib==0.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # chex # distrax @@ -299,95 +303,96 @@ jaxlib==0.4.38 # rlax jaxopt==0.8.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flask # torch kiwisolver==1.4.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 # werkzeug matplotlib==3.10.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # evosax # gymnax # seaborn mccabe==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flake8 # pylint mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py ml-collections==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax ml-dtypes==0.5.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jax # jaxlib # tensorstore mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy msgpack==1.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # orbax-checkpoint mujoco==3.2.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco-mjx mujoco-mjx==3.2.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax mypy-extensions==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black navix==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in nest-asyncio==1.6.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in # brax # chex # contourpy # distrax # dm-env + # dm-tree # evosax # flashbax # gym @@ -409,36 +414,36 @@ numpy==1.26.4 # tensorflow-probability # tensorstore # trimesh -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir opt-einsum==3.4.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jax optax==0.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in # brax # flax -orbax-checkpoint==0.11.0 +orbax-checkpoint==0.11.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # flax ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black # matplotlib # pytest @@ -446,99 +451,99 @@ packaging==24.2 # tensorboardx pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # seaborn pathspec==0.12.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # matplotlib # navix platformdirs==4.3.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black # pylint # wandb pluggy==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pytest protobuf==5.29.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint # tensorboardx # wandb psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir # wandb ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pycodestyle==2.12.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flake8 -pydantic==2.10.5 +pydantic==2.10.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb pydantic-core==2.27.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pydantic pyflakes==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flake8 pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pylint==3.3.3 +pylint==3.3.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix -pyopengl==3.1.7 +pyopengl==3.1.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # mujoco pyparsing==3.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib pytest==8.3.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # matplotlib # pandas pytinyrenderer==0.0.14 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # evosax # flax # gymnax @@ -548,25 +553,25 @@ pyyaml==6.0.2 # wandb reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flax # tyro # voir rlax==0.1.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix scipy==1.15.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # jax # jaxlib @@ -574,88 +579,88 @@ scipy==1.15.1 # mujoco-mjx seaborn==0.13.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gymnax sentry-sdk==2.20.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb setproctitle==1.3.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb setuptools-scm==8.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix shtab==1.7.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro simplejson==3.19.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # orbax-checkpoint six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # docker-pycreds # ml-collections # python-dateutil # tensorflow-probability smmap==5.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gitdb sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tensorboardx==2.6.2.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax tensorflow-probability==0.25.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # distrax tensorstore==0.1.71 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flashbax # flax # orbax-checkpoint tomli==2.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # black # pylint # pytest # setuptools-scm tomlkit==0.13.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pylint toolz==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # chex -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/purejaxrl/requirements.in -trimesh==4.5.3 +trimesh==4.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # brax # mujoco-mjx typeguard==4.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # astroid # black # brax @@ -674,39 +679,43 @@ typing-extensions==4.12.2 # typeguard # tyro # wandb -tyro==0.9.10 +tyro==0.9.13 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # sentry-sdk varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/purejaxrl/requirements.in -wandb==0.19.4 +wandb==0.19.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # navix werkzeug==3.1.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # flask +wrapt==1.17.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree zipp==3.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # etils # The following packages are considered to be unsafe in a requirements file: diff --git a/benchmarks/recursiongfn/requirements.cuda.txt b/benchmarks/recursiongfn/requirements.cuda.txt index af2df13b..3c1c81fe 100644 --- a/benchmarks/recursiongfn/requirements.cuda.txt +++ b/benchmarks/recursiongfn/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/recursiongfn/requirements.cuda.txt .pin/tmp-constraints-cuda-recursiongfn.txt benchmarks/recursiongfn/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -42,7 +42,7 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp @@ -50,11 +50,11 @@ blosc2==2.7.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tables -botorch==0.12.0 +botorch==0.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/recursiongfn/requirements.in -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -79,7 +79,7 @@ docker-pycreds==0.4.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # wandb -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -87,7 +87,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -112,12 +111,12 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -gpytorch==1.13 +gpytorch==1.14 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/recursiongfn/requirements.in # botorch -grpcio==1.69.0 +grpcio==1.70.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tensorboard @@ -142,7 +141,7 @@ jaxlib==0.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax -jaxtyping==0.2.19 +jaxtyping==0.2.37 # via # -c .pin/../.pin/constraints-cuda-torch.txt # gpytorch @@ -156,7 +155,7 @@ joblib==1.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # scikit-learn -linear-operator==0.5.3 +linear-operator==0.6 # via # -c .pin/../.pin/constraints-cuda-torch.txt # botorch @@ -186,7 +185,6 @@ ml-dtypes==0.5.1 mpmath==1.3.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt - # botorch # gpytorch # linear-operator # sympy @@ -203,6 +201,10 @@ multipledispatch==1.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # botorch +mypy-extensions==1.0.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # typing-inspect ndindex==1.9.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -223,7 +225,6 @@ numpy==1.26.4 # blosc2 # jax # jaxlib - # jaxtyping # ml-dtypes # numexpr # pandas @@ -235,57 +236,61 @@ numpy==1.26.4 # tensorboard # torch-geometric # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -298,9 +303,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -364,7 +371,7 @@ pyarrow==19.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/recursiongfn/requirements.in -pydantic==2.10.5 +pydantic==2.10.6 # via # -c .pin/../.pin/constraints-cuda-torch.txt # wandb @@ -380,6 +387,10 @@ pyparsing==3.2.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch-geometric +pyre-extensions==0.0.32 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # botorch pyro-api==0.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -393,7 +404,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -472,8 +483,9 @@ tensorboard-data-server==0.7.2 threadpoolctl==3.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt + # botorch # scikit-learn -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -503,28 +515,28 @@ tqdm==4.67.1 # -c .pin/../.pin/constraints-cuda-torch.txt # pyro-ppl # torch-geometric -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -typeguard==4.4.1 - # via - # -c .pin/../.pin/constraints-cuda-torch.txt - # jaxtyping typing-extensions==4.12.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # botorch - # jaxtyping # multidict # pydantic # pydantic-core + # pyre-extensions # reactivex # rich # tables # torch - # typeguard + # typing-inspect # wandb +typing-inspect==0.9.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # pyre-extensions tzdata==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -543,7 +555,11 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/recursiongfn/requirements.in -wandb==0.19.4 +wadler-lindig==0.1.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # jaxtyping +wandb==0.19.5 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/recursiongfn/requirements.in @@ -551,7 +567,7 @@ werkzeug==3.1.3 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tensorboard -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/recursiongfn/requirements.rocm.txt b/benchmarks/recursiongfn/requirements.rocm.txt index 3f8f2f44..0505cd61 100644 --- a/benchmarks/recursiongfn/requirements.rocm.txt +++ b/benchmarks/recursiongfn/requirements.rocm.txt @@ -5,200 +5,201 @@ # pip-compile --output-file=benchmarks/recursiongfn/requirements.rocm.txt .pin/tmp-constraints-rocm-recursiongfn.txt benchmarks/recursiongfn/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com absl-py==2.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp annotated-types==0.7.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pydantic antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp blosc2==2.7.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tables -botorch==0.12.0 +botorch==0.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # sentry-sdk charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests click==8.1.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera cvxopt==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in docker-pycreds==0.4.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch # torch-geometric gitdb==4.0.12 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gitpython gitpython==3.1.44 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # wandb giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -gpytorch==1.13 +gpytorch==1.14 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # botorch -grpcio==1.69.0 +grpcio==1.70.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl -jaxtyping==0.2.19 +jaxtyping==0.2.37 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gpytorch # linear-operator jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch # torch-geometric joblib==1.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # scikit-learn -linear-operator==0.5.3 +linear-operator==0.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # botorch # gpytorch markdown==3.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 # werkzeug mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # botorch + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gpytorch # linear-operator # sympy msgpack==1.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blosc2 multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl multipledispatch==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # botorch +mypy-extensions==1.0.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # typing-inspect ndindex==1.9.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blosc2 networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # torch numexpr==2.10.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blosc2 # tables numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blosc2 - # jaxtyping # numexpr # pandas # pyro-ppl @@ -208,135 +209,139 @@ numpy==1.26.4 # tables # tensorboard # torch-geometric -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # voir opt-einsum==3.4.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pyro-ppl ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tables # tensorboard pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rdkit platformdirs==4.3.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl protobuf==5.29.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard # wandb psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric # voir # wandb ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir py-cpuinfo==9.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # blosc2 # tables pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in -pydantic==2.10.5 +pydantic==2.10.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb pydantic-core==2.27.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pydantic pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich pyparsing==3.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric +pyre-extensions==0.0.32 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # botorch pyro-api==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pyro-ppl pyro-ppl==1.9.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # botorch python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf # wandb rdkit==2024.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch-geometric # wandb rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir scikit-learn==1.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gpytorch scipy==1.15.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # botorch # gpytorch @@ -346,117 +351,122 @@ scipy==1.15.1 # torch-sparse sentry-sdk==2.20.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb setproctitle==1.3.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # wandb six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # docker-pycreds # python-dateutil # tensorboard smmap==5.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gitdb sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tables==3.10.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in tensorboard==2.18.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in tensorboard-data-server==0.7.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard threadpoolctl==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # botorch # scikit-learn -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in # botorch # linear-operator # pyro-ppl torch-cluster==1.6.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in torch-geometric==2.6.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in torch-scatter==2.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in torch-sparse==0.6.18 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pyro-ppl # torch-geometric -typeguard==4.4.1 - # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # jaxtyping typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # botorch - # jaxtyping # multidict # pydantic # pydantic-core + # pyre-extensions # reactivex # rich # tables # torch - # typeguard + # typing-inspect # wandb -tzdata==2024.2 +typing-inspect==0.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # pyre-extensions +tzdata==2025.1 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # sentry-sdk varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/recursiongfn/requirements.in -wandb==0.19.4 +wadler-lindig==0.1.3 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # jaxtyping +wandb==0.19.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/recursiongfn/requirements.in werkzeug==3.1.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # The following packages are considered to be unsafe in a requirements file: diff --git a/benchmarks/rlhf/requirements.cuda.txt b/benchmarks/rlhf/requirements.cuda.txt index 5c2ec2bf..83796de4 100644 --- a/benchmarks/rlhf/requirements.cuda.txt +++ b/benchmarks/rlhf/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/rlhf/requirements.cuda.txt .pin/tmp-constraints-cuda-rlhf-gpus.txt benchmarks/rlhf/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -40,11 +40,11 @@ async-timeout==5.0.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -70,7 +70,7 @@ einops==0.8.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/rlhf/requirements.in -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -81,7 +81,6 @@ filelock==3.17.0 # huggingface-hub # torch # transformers - # triton frozenlist==1.5.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -98,7 +97,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # accelerate @@ -176,57 +175,61 @@ numpy==1.26.4 # scipy # transformers # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -239,9 +242,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -294,7 +299,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -347,7 +352,7 @@ tokenizers==0.21.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # transformers -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -360,16 +365,16 @@ tqdm==4.67.1 # datasets # huggingface-hub # transformers -transformers==4.48.1 +transformers==4.48.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/rlhf/requirements.in # trl -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -trl==0.13.0 +trl==0.14.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/rlhf/requirements.in @@ -398,7 +403,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/rlhf/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/rlhf/requirements.rocm.txt b/benchmarks/rlhf/requirements.rocm.txt index 2e6b3b2c..693c3150 100644 --- a/benchmarks/rlhf/requirements.rocm.txt +++ b/benchmarks/rlhf/requirements.rocm.txt @@ -5,214 +5,208 @@ # pip-compile --output-file=benchmarks/rlhf/requirements.rocm.txt .pin/tmp-constraints-rocm-rlhf-gpus.txt benchmarks/rlhf/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com -accelerate==1.2.1 +accelerate==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in # trl aiohappyeyeballs==2.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp aiohttp==3.11.11 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # fsspec aiosignal==1.3.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving async-timeout==5.0.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -attrs==24.3.0 +attrs==25.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera datasets==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in # trl dill==0.3.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # multiprocess -docstring-parser==0.16 - # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # tyro einops==0.8.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub - # pytorch-triton-rocm # torch # transformers frozenlist==1.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # aiosignal fsspec[http]==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # tokenizers # transformers idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests # yarl jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy multidict==6.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl multiprocess==0.70.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # pandas # transformers - # trl -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub # transformers pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets propcache==0.2.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp # yarl psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pyarrow==19.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # datasets # huggingface-hub @@ -220,106 +214,90 @@ pyyaml==6.0.2 # transformers reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving regex==2024.11.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # tyro + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # trl # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # accelerate # transformers -shtab==1.7.1 - # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # tyro six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tokenizers==0.21.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # transformers -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in # accelerate - # trl tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets # huggingface-hub # transformers -transformers==4.48.0 +transformers==4.48.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in # trl -trl==0.10.1 +trl==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/rlhf/requirements.in -typeguard==4.4.1 - # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # tyro typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # multidict # reactivex # rich # torch - # typeguard - # tyro -tyro==0.9.10 - # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # trl -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/rlhf/requirements.in xxhash==3.5.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # datasets yarl==1.18.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # aiohttp diff --git a/benchmarks/timm/requirements.cuda.txt b/benchmarks/timm/requirements.cuda.txt index 14635500..d85474e1 100644 --- a/benchmarks/timm/requirements.cuda.txt +++ b/benchmarks/timm/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/timm/requirements.cuda.txt .pin/tmp-constraints-cuda-timm.txt benchmarks/timm/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -18,7 +18,7 @@ asttokens==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # giving -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -30,7 +30,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -39,7 +39,6 @@ filelock==3.17.0 # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -50,7 +49,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/timm/requirements.in @@ -112,57 +111,61 @@ numpy==1.26.4 # scipy # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -175,9 +178,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -244,14 +249,14 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/timm/requirements.in # torchvision # xformers -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/timm/requirements.in @@ -259,7 +264,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -283,7 +288,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/timm/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/timm/requirements.rocm.txt b/benchmarks/timm/requirements.rocm.txt index 4cd62365..a6debea1 100644 --- a/benchmarks/timm/requirements.rocm.txt +++ b/benchmarks/timm/requirements.rocm.txt @@ -5,177 +5,176 @@ # pip-compile --output-file=benchmarks/timm/requirements.rocm.txt .pin/tmp-constraints-rocm-timm.txt benchmarks/timm/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub - # pytorch-triton-rocm # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/timm/requirements.in idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/timm/requirements.in # huggingface-hub # omegaconf reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/timm/requirements.in sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/timm/requirements.in # torchvision -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/timm/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # reactivex # rich # torch urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/timm/requirements.in diff --git a/benchmarks/torchatari/requirements.cuda.txt b/benchmarks/torchatari/requirements.cuda.txt index 90f19eb1..f9e44607 100644 --- a/benchmarks/torchatari/requirements.cuda.txt +++ b/benchmarks/torchatari/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/torchatari/requirements.cuda.txt .pin/tmp-constraints-cuda-torchatari.txt benchmarks/torchatari/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -14,6 +14,7 @@ absl-py==2.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # dm-env + # dm-tree # tensorboard antlr4-python3-runtime==4.9.3 # via @@ -27,6 +28,10 @@ asttokens==3.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # giving +attrs==25.1.0 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree cantilever==0.1.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -44,7 +49,7 @@ dm-env==1.6 # via # -c .pin/../.pin/constraints-cuda-torch.txt # envpool -dm-tree==0.1.8 +dm-tree==0.1.9 # via # -c .pin/../.pin/constraints-cuda-torch.txt # dm-env @@ -56,7 +61,7 @@ envpool==0.8.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchatari/requirements.in -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -68,7 +73,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -78,7 +82,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -grpcio==1.69.0 +grpcio==1.70.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tensorboard @@ -95,7 +99,7 @@ gymnasium==1.0.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # envpool -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # cantilever @@ -155,6 +159,7 @@ numpy==1.26.4 # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchatari/requirements.in # dm-env + # dm-tree # envpool # gym # gymnasium @@ -164,57 +169,61 @@ numpy==1.26.4 # scipy # tensorboard # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -227,9 +236,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -308,7 +319,7 @@ tensorboard-data-server==0.7.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tensorboard -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -319,7 +330,7 @@ torchcompat==1.1.4 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/torchatari/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -342,7 +353,7 @@ typing-extensions==4.12.2 # torch # typeguard # tyro -tyro==0.9.12 +tyro==0.9.13 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchatari/requirements.in @@ -359,7 +370,11 @@ werkzeug==3.1.3 # via # -c .pin/../.pin/constraints-cuda-torch.txt # tensorboard -xformers==0.0.29.post1 +wrapt==1.17.2 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # dm-tree +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/torchatari/requirements.rocm.txt b/benchmarks/torchatari/requirements.rocm.txt index 8ec1a917..52fb1e96 100644 --- a/benchmarks/torchatari/requirements.rocm.txt +++ b/benchmarks/torchatari/requirements.rocm.txt @@ -5,231 +5,236 @@ # pip-compile --output-file=benchmarks/torchatari/requirements.rocm.txt .pin/tmp-constraints-rocm-torchatari.txt benchmarks/torchatari/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com absl-py==2.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # dm-env + # dm-tree # tensorboard antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf appdirs==1.4.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # cantilever asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving +attrs==25.1.0 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree cantilever==0.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in cloudpickle==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym # gymnasium codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera dm-env==1.6 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool -dm-tree==0.1.8 +dm-tree==0.1.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # dm-env docstring-parser==0.16 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro envpool==0.8.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname farama-notifications==0.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gymnasium -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -grpcio==1.69.0 +grpcio==1.70.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard gym==0.26.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in # envpool gym-notices==0.0.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # gym gymnasium==1.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # cantilever # torchcompat jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown==3.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 # werkzeug mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in # dm-env + # dm-tree # envpool # gym # gymnasium # tensorboard -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir optree==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool # tensorboard protobuf==5.29.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro # voir shtab==1.7.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch tensorboard==2.18.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in tensorboard-data-server==0.7.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in torchcompat==1.1.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchatari/requirements.in typeguard==4.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tyro types-protobuf==5.29.1.20241207 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # envpool # gymnasium # optree @@ -238,23 +243,27 @@ typing-extensions==4.12.2 # torch # typeguard # tyro -tyro==0.9.10 +tyro==0.9.13 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchatari/requirements.in varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchatari/requirements.in werkzeug==3.1.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # tensorboard +wrapt==1.17.2 + # via + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # dm-tree # The following packages are considered to be unsafe in a requirements file: # setuptools diff --git a/benchmarks/torchvision/requirements.cuda.txt b/benchmarks/torchvision/requirements.cuda.txt index 4090e7b9..e9941064 100644 --- a/benchmarks/torchvision/requirements.cuda.txt +++ b/benchmarks/torchvision/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/torchvision/requirements.cuda.txt .pin/tmp-constraints-cuda-torchvision.txt benchmarks/torchvision/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -22,7 +22,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -30,7 +30,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -40,7 +39,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchcompat @@ -98,57 +97,61 @@ numpy==1.26.4 # scipy # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -161,9 +164,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -216,7 +221,7 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -228,7 +233,7 @@ torchcompat==1.1.4 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/torchvision/requirements.in -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchvision/requirements.in @@ -236,7 +241,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchvision/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -255,7 +260,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/torchvision/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/torchvision/requirements.rocm.txt b/benchmarks/torchvision/requirements.rocm.txt index 7d746afb..351b08fd 100644 --- a/benchmarks/torchvision/requirements.rocm.txt +++ b/benchmarks/torchvision/requirements.rocm.txt @@ -5,149 +5,148 @@ # pip-compile --output-file=benchmarks/torchvision/requirements.rocm.txt .pin/tmp-constraints-rocm-torchvision.txt benchmarks/torchvision/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchcompat jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision/requirements.in # torchvision torchcompat==1.1.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchvision/requirements.in -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # reactivex # rich # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchvision/requirements.in diff --git a/benchmarks/torchvision_ddp/requirements.cuda.txt b/benchmarks/torchvision_ddp/requirements.cuda.txt index 60a3f9f8..93127f5f 100644 --- a/benchmarks/torchvision_ddp/requirements.cuda.txt +++ b/benchmarks/torchvision_ddp/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/torchvision_ddp/requirements.cuda.txt .pin/tmp-constraints-cuda-torchvision.txt benchmarks/torchvision_ddp/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -22,7 +22,7 @@ codefind==0.1.7 # via # -c .pin/../.pin/constraints-cuda-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -30,7 +30,6 @@ filelock==3.17.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -40,7 +39,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torchcompat @@ -98,57 +97,61 @@ numpy==1.26.4 # scipy # torchvision # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -161,9 +164,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -216,7 +221,7 @@ sympy==1.13.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -228,7 +233,7 @@ torchcompat==1.1.4 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/torchvision_ddp/requirements.in -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchvision_ddp/requirements.in @@ -236,7 +241,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/torchvision_ddp/requirements.in -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -255,7 +260,7 @@ voir==0.2.19 # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt # -r benchmarks/torchvision_ddp/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/torchvision_ddp/requirements.rocm.txt b/benchmarks/torchvision_ddp/requirements.rocm.txt index 5e54bb9d..8f2a0d2e 100644 --- a/benchmarks/torchvision_ddp/requirements.rocm.txt +++ b/benchmarks/torchvision_ddp/requirements.rocm.txt @@ -5,149 +5,148 @@ # pip-compile --output-file=benchmarks/torchvision_ddp/requirements.rocm.txt .pin/tmp-constraints-rocm-torchvision.txt benchmarks/torchvision_ddp/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # pytorch-triton-rocm + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -importlib-resources==6.5.2 +importlib-resources==5.13.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchcompat jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision_ddp/requirements.in # torchvision torchcompat==1.1.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchvision_ddp/requirements.in -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision_ddp/requirements.in tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/torchvision_ddp/requirements.in typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # reactivex # rich # torch varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/torchvision_ddp/requirements.in diff --git a/benchmarks/vjepa/requirements.cuda.txt b/benchmarks/vjepa/requirements.cuda.txt index 9dcbe4b3..c078b56c 100644 --- a/benchmarks/vjepa/requirements.cuda.txt +++ b/benchmarks/vjepa/requirements.cuda.txt @@ -5,7 +5,7 @@ # pip-compile --output-file=benchmarks/vjepa/requirements.cuda.txt .pin/tmp-constraints-cuda-vjepa-gpus.txt benchmarks/vjepa/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html --find-links https://data.pyg.org/whl/torch-2.5.1+cu121.html --trusted-host pypi.ngc.nvidia.com @@ -27,7 +27,7 @@ braceexpand==0.1.7 # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/vjepa/requirements.in # webdataset -certifi==2024.12.14 +certifi==2025.1.31 # via # -c .pin/../.pin/constraints-cuda-torch.txt # requests @@ -51,7 +51,7 @@ einops==0.8.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/vjepa/requirements.in -executing==2.1.0 +executing==2.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # varname @@ -60,7 +60,6 @@ filelock==3.17.0 # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub # torch - # triton fsspec==2024.9.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt @@ -71,7 +70,7 @@ giving==0.4.3 # -c .pin/../.pin/constraints-cuda-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # timm @@ -138,57 +137,61 @@ numpy==1.26.4 # torchvision # webdataset # xformers -nvidia-cublas-cu12==12.1.3.1 +nvidia-cublas-cu12==12.6.4.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cudnn-cu12 # nvidia-cusolver-cu12 # torch -nvidia-cuda-cupti-cu12==12.1.105 +nvidia-cuda-cupti-cu12==12.6.80 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cuda-nvcc-cu12==12.6.85 +nvidia-cuda-nvcc-cu12==12.8.61 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin -nvidia-cuda-nvrtc-cu12==12.1.105 +nvidia-cuda-nvrtc-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cuda-runtime-cu12==12.1.105 +nvidia-cuda-runtime-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cudnn-cu12==9.1.0.70 +nvidia-cudnn-cu12==9.5.1.17 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cufft-cu12==11.0.2.54 +nvidia-cufft-cu12==11.3.0.4 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-curand-cu12==10.3.2.106 +nvidia-curand-cu12==10.3.7.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch -nvidia-cusolver-cu12==11.4.5.107 +nvidia-cusolver-cu12==11.7.1.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # torch -nvidia-cusparse-cu12==12.1.0.106 +nvidia-cusparse-cu12==12.5.4.2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin # nvidia-cusolver-cu12 # torch -nvidia-ml-py==12.560.30 +nvidia-cusparselt-cu12==0.6.3 + # via + # -c .pin/../.pin/constraints-cuda-torch.txt + # torch +nvidia-ml-py==12.570.86 # via # -c .pin/../.pin/constraints-cuda-torch.txt # voir @@ -201,9 +204,11 @@ nvidia-nvjitlink-cu12==12.6.85 # via # -c .pin/../.pin/constraints-cuda-torch.txt # jax-cuda12-plugin + # nvidia-cufft-cu12 # nvidia-cusolver-cu12 # nvidia-cusparse-cu12 -nvidia-nvtx-cu12==12.1.105 + # torch +nvidia-nvtx-cu12==12.6.77 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -251,7 +256,7 @@ python-dateutil==2.9.0.post0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas -pytz==2024.2 +pytz==2025.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # pandas @@ -300,7 +305,7 @@ timm==1.0.14 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/vjepa/requirements.in -torch==2.5.1+cu121 +torch==2.6.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -c .pin/../constraints/cuda.txt @@ -308,7 +313,7 @@ torch==2.5.1+cu121 # timm # torchvision # xformers -torchvision==0.20.1+cu121 +torchvision==0.21.0+cu126 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/vjepa/requirements.in @@ -317,7 +322,7 @@ tqdm==4.67.1 # via # -c .pin/../.pin/constraints-cuda-torch.txt # huggingface-hub -triton==3.1.0 +triton==3.2.0 # via # -c .pin/../.pin/constraints-cuda-torch.txt # torch @@ -350,7 +355,7 @@ webdataset==0.2.100 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r benchmarks/vjepa/requirements.in -xformers==0.0.29.post1 +xformers==0.0.29.post2 # via # -c .pin/../.pin/constraints-cuda-torch.txt # -r .pin/../constraints/extra/torch.cuda.txt diff --git a/benchmarks/vjepa/requirements.rocm.txt b/benchmarks/vjepa/requirements.rocm.txt index ee7d650d..b3282638 100644 --- a/benchmarks/vjepa/requirements.rocm.txt +++ b/benchmarks/vjepa/requirements.rocm.txt @@ -5,166 +5,165 @@ # pip-compile --output-file=benchmarks/vjepa/requirements.rocm.txt .pin/tmp-constraints-rocm-vjepa-gpus.txt benchmarks/vjepa/requirements.in # --extra-index-url https://pypi.ngc.nvidia.com ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 --trusted-host pypi.ngc.nvidia.com antlr4-python3-runtime==4.9.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # omegaconf asttokens==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving beartype==0.19.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in braceexpand==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in # webdataset -certifi==2024.12.14 +certifi==2025.1.31 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests charset-normalizer==3.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests cloudpickle==3.1.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # submitit codefind==0.1.7 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera decord==0.6.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in einops==0.8.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in -executing==2.1.0 +executing==2.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # varname -filelock==3.16.1 +filelock==3.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub - # pytorch-triton-rocm # torch fsspec==2024.9.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # torch giving==0.4.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # ptera # voir -huggingface-hub==0.27.1 +huggingface-hub==0.28.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # timm idna==3.10 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests jinja2==3.1.5 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch markdown-it-py==3.0.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich markupsafe==3.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # jinja2 mdurl==0.1.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # markdown-it-py mpmath==1.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # sympy networkx==3.4.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch numpy==1.26.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in # decord # opencv-python # pandas # torchvision # webdataset -nvidia-ml-py==12.560.30 +nvidia-ml-py==12.570.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir omegaconf==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir opencv-python==4.11.0.86 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in ovld==0.3.9 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir packaging==24.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub pandas==2.2.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in pillow==11.1.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torchvision psutil==5.9.8 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir ptera==1.4.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir pygments==2.19.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # rich python-dateutil==2.9.0.post0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas -pytorch-triton-rocm==3.1.0 +pytorch-triton-rocm==3.2.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -pytz==2024.2 +pytz==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas pyyaml==6.0.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in # huggingface-hub # omegaconf @@ -172,77 +171,77 @@ pyyaml==6.0.2 # webdataset reactivex==4.0.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving requests==2.32.3 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub rich==13.9.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # voir safetensors==0.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # timm six==1.17.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # python-dateutil submitit==1.5.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in sympy==1.13.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # torch -timm==1.0.13 +timm==1.0.14 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in -torch==2.5.1+rocm6.2 +torch==2.6.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in # timm # torchvision -torchvision==0.20.1+rocm6.2 +torchvision==0.21.0+rocm6.2.4 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in # timm tqdm==4.67.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub typing-extensions==4.12.2 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # huggingface-hub # reactivex # rich # submitit # torch -tzdata==2024.2 +tzdata==2025.1 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # pandas urllib3==2.3.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # requests varname==0.14.0 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # giving voir==0.2.19 # via - # -c .pin/../.pin/constraints-rocm-torch.txt - # -c .pin/../constraints/rocm.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/constraints/rocm.txt # -r benchmarks/vjepa/requirements.in webdataset==0.2.100 # via - # -c .pin/../.pin/constraints-rocm-torch.txt + # -c /network/scratch/d/delaunap/shared/milabench/.pin/constraints-rocm-torch.txt # -r benchmarks/vjepa/requirements.in diff --git a/constraints/cuda.txt b/constraints/cuda.txt index 0df36b52..9e2bc20e 100644 --- a/constraints/cuda.txt +++ b/constraints/cuda.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu126 # # diff --git a/constraints/rocm.txt b/constraints/rocm.txt index aeaf8c6f..5c7283c4 100644 --- a/constraints/rocm.txt +++ b/constraints/rocm.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/rocm6.2 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 # # diff --git a/milabench/_version.py b/milabench/_version.py index 71d24764..eccaee6a 100644 --- a/milabench/_version.py +++ b/milabench/_version.py @@ -1,5 +1,5 @@ """This file is generated, do not modify""" -__tag__ = "v1.0.0-24-g776e3e1" -__commit__ = "776e3e1bba8be548c22fd4a811092e7e00128689" -__date__ = "2025-01-16 15:14:33 -0500" +__tag__ = "v1.0.0-48-g61ca0a3" +__commit__ = "61ca0a328fe57ed5aca60c11506dc57e85a8acbb" +__date__ = "2025-02-04 12:11:28 -0500" From 6c98d05650b9c5a8c0de4c7f6a2abdc49c1acbb3 Mon Sep 17 00:00:00 2001 From: "pierre.delaunay" Date: Thu, 6 Feb 2025 13:13:08 -0500 Subject: [PATCH 2/3] Tweak memory extract to be use batch increment instead of an absolute batch size --- config/base.yaml | 4 +- config/examples/system.yaml | 6 +-- config/scaling/L40S.yaml | 13 +---- milabench/sizer.py | 102 ++++++++++++++++++++++++------------ milabench/system.py | 3 ++ 5 files changed, 78 insertions(+), 50 deletions(-) diff --git a/config/base.yaml b/config/base.yaml index 82ff3a9a..a8ce08f8 100644 --- a/config/base.yaml +++ b/config/base.yaml @@ -216,7 +216,7 @@ resnet50-noio: inherits: _torchvision voir: options: - stop: 500 + stop: 60 interval: "1s" tags: @@ -229,7 +229,7 @@ resnet50-noio: argv: --model: resnet50 - --batch-size: batch_resize(256) + --batch-size: auto_batch(256) --loader: synthetic_fixed --optim: channel_last diff --git a/config/examples/system.yaml b/config/examples/system.yaml index 06f103cb..302becc9 100644 --- a/config/examples/system.yaml +++ b/config/examples/system.yaml @@ -41,14 +41,14 @@ system: multirun: runs: # Force batch size to populate the sizing model - - name: "bs{sizer.batch_size}" + - name: "bs{sizer.batch_size}.{time}" matrix: sizer.auto: 1 sizer.batch_size: [1, 2, 4, 8, 16, 32, 64, 128] sizer.save: ["scaling.yaml"] # Matrix run - - name: "c{sizer.capacity}_m{sizer.multiple}_w{cpu.n_workers}" + - name: "c{sizer.capacity}_m{sizer.multiple}_w{cpu.n_workers}.{time}" matrix: cpu.auto: 1 cpu.n_workers: [2, 4, 8, 16, 32] @@ -58,7 +58,7 @@ multirun: sizer.save: ["scaling.yaml"] # Auto run - - name: "auto" + - name: "auto.{time}" matrix: cpu.auto: 1 sizer.auto: 1 diff --git a/config/scaling/L40S.yaml b/config/scaling/L40S.yaml index 210903fb..29b34aac 100644 --- a/config/scaling/L40S.yaml +++ b/config/scaling/L40S.yaml @@ -1,13 +1,4 @@ -resnet152-ddp-gpus: - observations: - - {batch_size: 0, cpu: 8, memory: 25995 MiB, perf: 1995.03} resnet50: observations: - - {batch_size: 1, cpu: 8, memory: 46041 MiB, perf: 78.92} - - {batch_size: 2, cpu: 8, memory: 1237 MiB, perf: 159.82} - - {batch_size: 2, cpu: 8, memory: 34611 MiB, perf: 161.62} - - {batch_size: 4, cpu: 8, memory: 1331 MiB, perf: 315.69} - - {batch_size: 4, cpu: 8, memory: 46041 MiB, perf: 308.07} - - {batch_size: 8, cpu: 8, memory: 1489 MiB, perf: 603.96} - - {batch_size: 8, cpu: 8, memory: 46041 MiB, perf: 602.72} -version: 2.0 + - {batch_size: 248, cpu: 8, memory: 12798 MiB, perf: 654.31} +version: 2 diff --git a/milabench/sizer.py b/milabench/sizer.py index 0d28fe45..e0ef6dca 100644 --- a/milabench/sizer.py +++ b/milabench/sizer.py @@ -9,6 +9,7 @@ from voir.instruments.gpu import get_gpu_info from cantilever.core.statstream import StatStream +from .syslog import syslog from .system import CPUOptions, SizerOptions, system_global, option from .validation.validation import ValidationLayer @@ -273,7 +274,7 @@ def __init__(self): self.filepath = option("sizer.save", str, None) sizer = Sizer() - if os.path.exists(self.filepath): + if self.filepath and os.path.exists(self.filepath): with open(self.filepath, "r") as fp: self.memory = yaml.safe_load(fp) or {} else: @@ -286,7 +287,9 @@ def __init__(self): self.scaling = None self.stats = defaultdict(lambda: StatStream(drop_first_obs=0)) self.benchname = None - self.batch_size = 0 + self.batch_size = None + self.active_count = defaultdict(int) + self.rc = defaultdict(int) self.max_usage = float("-inf") # Usage from the gpu monitor self.peak_usage = float("-inf") # Usage provided by the bench itself (for jax) self.early_stopped = False @@ -302,6 +305,7 @@ def on_cpu_count_set(self, pack, _, value): self.stats["cpu"] += value def on_batch_size_set(self, pack, _, value): + self.batch_size = value self.stats["batch_size"] += value def convert(self): @@ -322,11 +326,13 @@ def convert(self): def on_start(self, entry): if self.filepath is None: return - + self.benchname = entry.pack.config["name"] self.max_usage = float("-inf") self.peak_usage = float("-inf") + self.active_count[self.benchname] += 1 + def on_data(self, entry): if self.filepath is None: return @@ -359,46 +365,71 @@ def on_end(self, entry): if self.filepath is None: return - if ( - self.benchname is None - or self.batch_size is None - or self.max_memory_usage() == float("-inf") - ): + if self.benchname is None: + syslog("MemoryUsageExtractor: Skipping missing benchmark {}", entry) + return + + if self.batch_size is None: + syslog("MemoryUsageExtractor: Skipping missing batch_size {}", entry) + return + + if self.max_memory_usage() == float("-inf"): + syslog("MemoryUsageExtractor: Missing memory info {}", entry) return # Only update is successful rc = entry.data["return_code"] if rc == 0 or self.early_stopped: - config = self.memory.setdefault(self.benchname, dict()) - observations = config.setdefault("observations", []) - - obs = { - "cpu": int(self.stats["cpu"].avg), - "batch_size": int(self.stats["batch_size"].avg), - "memory": f"{int(self.stats['memory'].max)} MiB", - "perf": float(f"{self.stats['perf'].avg:.2f}"), - } - - if memorypeak := self.stats.pop("memorypeak", None): - if memorypeak.current_count != 0: - obs["memory"] = f"{int(memorypeak.max)} MiB", - - observations.append(obs) - - config["observations"] = list(sorted(observations, key=lambda x: x["batch_size"])) - - self.benchname = None - self.batch_size = None - self.stats = defaultdict(lambda: StatStream(drop_first_obs=0)) - self.max_usage = float("-inf") - self.peak_usage = float("-inf") - - def report(self, *args): + rc = 0 + + self.rc[self.benchname] += rc + self.active_count[self.benchname] -= 1 + + if self.active_count[self.benchname] <= 0: + + if self.rc[self.benchname] == 0: + self.push_observation() + else: + syslog("MemoryUsageExtractor: Could not add scaling data because of a failure") + + self.benchname = None + self.batch_size = None + self.stats = defaultdict(lambda: StatStream(drop_first_obs=0)) + self.max_usage = float("-inf") + self.peak_usage = float("-inf") + # avoid losing results + try: + self.save() + except Exception as err: + print(f"MemoryUsageExtractor: Could not save scaling file because of {err}") + + def push_observation(self): + config = self.memory.setdefault(self.benchname, dict()) + observations = config.setdefault("observations", []) + + obs = { + "cpu": int(self.stats["cpu"].avg), + "batch_size": int(self.stats["batch_size"].avg), + "memory": f"{int(self.stats['memory'].max)} MiB", + "perf": float(f"{self.stats['perf'].avg:.2f}"), + } + + if memorypeak := self.stats.pop("memorypeak", None): + if memorypeak.current_count != 0: + obs["memory"] = f"{int(memorypeak.max)} MiB", + + observations.append(obs) + config["observations"] = list(sorted(observations, key=lambda x: x["batch_size"])) + + def save(self): if self.filepath is not None: with open(self.filepath, "w") as file: yaml.dump(self.memory, file, Dumper=compact_dump()) + def report(self, *args): + self.save() + def arch_to_device(arch): device_types = [ @@ -462,7 +493,10 @@ def batch_resize(default): newvalue = default if gpu_opt.enabled: - newvalue = suggested_batch_size(pack) + if gpu_opt.add is not None and isinstance(gpu_opt.add, int): + newvalue = max(1, default + gpu_opt.add) + else: + newvalue = suggested_batch_size(pack) broadcast(on_batch_size_set, pack, default, newvalue) return newvalue diff --git a/milabench/system.py b/milabench/system.py index 041a23c0..e568d9d9 100644 --- a/milabench/system.py +++ b/milabench/system.py @@ -184,6 +184,9 @@ class SizerOptions: # overrides the batch size to use for all benchmarks size: int = defaultfield("sizer.batch_size", int, None) + # Add a fixed number to the current batch size + add: int = defaultfield("sizer.batch_size_add", int, None) + # Enables auto batch resize autoscale: bool = defaultfield("sizer.auto", int, 0) From 52430af13389b03548e6cbba5d25c3e9a1078d96 Mon Sep 17 00:00:00 2001 From: "pierre.delaunay" Date: Thu, 6 Feb 2025 18:18:54 -0500 Subject: [PATCH 3/3] L40S scaling runs --- config/base.yaml | 65 ++++++++++++------------ config/scaling/L40S.yaml | 105 ++++++++++++++++++++++++++++++++++++++- milabench/sizer.py | 4 +- 3 files changed, 140 insertions(+), 34 deletions(-) diff --git a/config/base.yaml b/config/base.yaml index a8ce08f8..e98edc6e 100644 --- a/config/base.yaml +++ b/config/base.yaml @@ -148,7 +148,7 @@ _accelerate_opt: --dataset_config_name: "wikitext-103-v1" --dataset_rev: "b08601e" --validation_split_percentage: 5 - --per_gpu_batch_size: 1 + --per_gpu_batch_size: auto_batch(1) --cpus_per_gpu: "auto({n_worker}, 8)" --cache: "{milabench_cache}" @@ -195,7 +195,7 @@ resnet50: voir: options: stop: 60 - gpu_poll: 0.1 + gpu_poll: 0.5 inherits: _torchvision tags: @@ -244,7 +244,7 @@ resnet152-ddp-gpus: argv: --model: resnet152 - --batch-size: 256 + --batch-size: auto_batch(256) --num-workers: "auto({n_worker}, 8)" --loader: torch @@ -258,7 +258,7 @@ _convnext_large-base: - monogpu argv: --model: convnext_large - --batch-size: 128 + --batch-size: auto_batch(128) voir: options: stop: 30 @@ -294,7 +294,7 @@ regnet_y_128gf: - monogpu argv: --model: regnet_y_128gf - --batch-size: 64 + --batch-size: auto_batch(64) _bert-base: inherits: _hf @@ -308,7 +308,7 @@ _bert-base: - monogpu argv: --model: "Bert" - --batch-size: 32 + --batch-size: auto_batch(32) voir: options: stop: 30 @@ -344,7 +344,7 @@ t5: - monogpu argv: --model: "T5" - --batch-size: 16 + --batch-size: auto_batch(16) reformer: inherits: _hf @@ -357,7 +357,7 @@ reformer: - monogpu argv: --model: "Reformer" - --batch-size: 32 + --batch-size: auto_batch(32) whisper: inherits: _hf @@ -368,7 +368,7 @@ whisper: - monogpu argv: --model: "Whisper" - --batch-size: 64 + --batch-size: auto_batch(64) focalnet: inherits: _timm @@ -381,6 +381,7 @@ focalnet: method: per_gpu argv: --model: focalnet_base_lrf + --batch-size: auto_batch(128) brax: # Brax requires very specific sizes to work @@ -400,7 +401,7 @@ brax: n: 1 argv: --episode-length: 20 - --batch-size: 1024 + --batch-size: auto_batch(1024) --num-minibatches: 32 --num-envs: 8192 @@ -415,7 +416,7 @@ _diffusion: argv: --num_epochs: 5 - --batch_size: 32 + --batch_size: auto_batch(32) --num_workers: "auto({n_worker}, 8)" --cache: "{milabench_cache}" @@ -459,7 +460,7 @@ _lightning: --loader: pytorch --data: "{milabench_data}/FakeImageNet" --model: resnet152 - --batch-size: 256 + --batch-size: auto_batch(256) lightning: inherits: _lightning @@ -504,7 +505,7 @@ dinov2-giant-single: --config-file: "{benchmark_folder}/src/dinov2/configs/train/vitg14.yaml" # THOSE NEED TO BE LAST train.dataset_path=ImageNet:split=TRAIN:root={milabench_data}/FakeImageNet:extra={milabench_data}/FakeImageNet: true - train.batch_size_per_gpu=32: true + train.batch_size_per_gpu=auto_batch(32): true train.saveckp_freq=100: true train.num_workers=10: true @@ -519,7 +520,7 @@ dinov2-giant-gpus: --config-file: "{benchmark_folder}/src/dinov2/configs/train/vitg14.yaml" # THOSE NEED TO BE LAST train.dataset_path=ImageNet:split=TRAIN:root={milabench_data}/FakeImageNet:extra={milabench_data}/FakeImageNet: true - train.batch_size_per_gpu=32: true + train.batch_size_per_gpu=auto_batch(32): true train.saveckp_freq=100: true train.num_workers=10: true @@ -535,7 +536,7 @@ dinov2-giant-nodes: --config-file: "{benchmark_folder}/src/dinov2/configs/train/vitg14.yaml" # THOSE NEED TO BE LAST train.dataset_path=ImageNet:split=TRAIN:root={milabench_data}/FakeImageNet:extra={milabench_data}/FakeImageNet: true - train.batch_size_per_gpu=12: true + train.batch_size_per_gpu=auto_batch(12): true train.saveckp_freq=100: true train.num_workers=10: true @@ -576,7 +577,7 @@ llm-lora-single: checkpointer.checkpoint_dir={milabench_data}/llama3_8B: true metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-8B": true - batch_size=8: true + batch_size=auto_batch(8): true gradient_accumulation_steps=8: true device={device_name}: true @@ -602,7 +603,7 @@ llm-lora-ddp-gpus: safetensors=true: true metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-8B": true - batch_size=8: true + batch_size=auto_batch(8): true gradient_accumulation_steps=8: true device={device_name}: true @@ -629,7 +630,7 @@ llm-lora-ddp-nodes: safetensors=true: true metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-8B": true - batch_size=8: true + batch_size=auto_batch(8): true gradient_accumulation_steps=8: true device={device_name}: true num_machines: 2 @@ -656,7 +657,7 @@ llm-lora-mp-gpus: safetensors=true: true metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-70B": true - batch_size=8: true + batch_size=auto_batch(8): true gradient_accumulation_steps=1: true device={device_name}: true @@ -685,7 +686,7 @@ llm-full-mp-gpus: metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-70B": true safetensors=true: true - batch_size=2: true + batch_size=auto_batch(2): true gradient_accumulation_steps=1: true device={device_name}: true @@ -710,7 +711,7 @@ llm-full-mp-nodes: metric_logger.log_dir={milabench_extra}/metrics: true repo_id="meta-llama/Meta-Llama-3.1-70B": true safetensors=true: true - batch_size=2: true + batch_size=auto_batch(2): true gradient_accumulation_steps=1: true device={device_name}: true @@ -736,7 +737,7 @@ dqn: dqn: true --num_envs: auto({cpu_per_gpu}, 128) --buffer_size: 131072 - --buffer_batch_size: 65536 + --buffer_batch_size: auto_batch(65536) --env_name: CartPole-v1 --training_interval: 10 @@ -746,7 +747,7 @@ ppo: ppo: true --num_envs: auto({cpu_per_gpu}, 128) --num_steps: 10 - --num_minibatches: 32 + --num_minibatches: auto_batch(32) --update_epochs: 4 --env_name: hopper --total_timesteps: 2000000 @@ -766,7 +767,7 @@ pna: argv: --model: 'PNA' --num-samples: 100000 - --batch-size: 4096 + --batch-size: auto_batch(4096) --num-workers: "auto({n_worker}, 0)" dimenet: @@ -775,7 +776,7 @@ dimenet: --model: 'DimeNet' --num-samples: 100000 --use3d: True - --batch-size: 16 + --batch-size: auto_batch(16) --num-workers: "auto({n_worker}, 0)" recursiongfn: @@ -789,7 +790,7 @@ recursiongfn: method: per_gpu argv: - --batch_size: 128 + --batch_size: auto_batch(128) --num_workers: "auto({n_worker}, 8)" --num_steps: 100 --layer_width: 128 @@ -806,7 +807,7 @@ torchatari: - monogpu - gym argv: - --num-minibatches: 16 + --num-minibatches: auto_batch(16) --update-epochs: 4 --num-steps: 128 --num-envs: auto({cpu_per_gpu}, 128) @@ -823,7 +824,7 @@ _llava: tags: - llm argv: - --batch_size: 1 + --batch_size: auto_batch(1) --num_workers: "auto({n_worker}, 4)" --gradient_accumulation_steps: 1 @@ -834,7 +835,7 @@ llava-single: plan: method: per_gpu argv: - --batch_size: 1 + --batch_size: auto_batch(1) --num_workers: "auto({n_worker}, 4)" --gradient_accumulation_steps: 1 @@ -846,7 +847,7 @@ llava-gpus: method: njobs n: 1 argv: - --batch_size: 1 + --batch_size: auto_batch(1) --num_workers: "auto({n_worker}, 4)" --gradient_accumulation_steps: 1 @@ -865,7 +866,7 @@ _rlhf: --output_dir: "{milabench_extra}/output" --model_name_or_path: EleutherAI/pythia-1b-deduped - --per_device_train_batch_size: 64 + --per_device_train_batch_size: auto_batch(64) --logging_strategy: "no" --log_level: "critical" --bf16: true @@ -894,7 +895,7 @@ _vjepa: tags: - video argv: - --batch_size: 24 + --batch_size: auto_batch(24) --num_workers: "auto({n_worker}, 12)" --dataset: "{milabench_data}/FakeVideo/video_metainfo.csv" --output: "{milabench_extra}" diff --git a/config/scaling/L40S.yaml b/config/scaling/L40S.yaml index 29b34aac..3d7bf32e 100644 --- a/config/scaling/L40S.yaml +++ b/config/scaling/L40S.yaml @@ -1,4 +1,107 @@ +bert-fp16: + observations: + - {batch_size: 24, cpu: 8, memory: 11938 MiB, perf: 85.28, time: 1738868625.0704234} + - {batch_size: 40, cpu: 8, memory: 18916 MiB, perf: 88.82, time: 1738877412.8066187} +bert-fp32: + observations: + - {batch_size: 24, cpu: 8, memory: 15726 MiB, perf: 24.05, time: 1738868555.9294276} + - {batch_size: 40, cpu: 8, memory: 25386 MiB, perf: 24.04, time: 1738877325.9952302} +bert-tf32: + observations: + - {batch_size: 24, cpu: 8, memory: 15726 MiB, perf: 24.05, time: 1738868754.2212114} + - {batch_size: 40, cpu: 8, memory: 25386 MiB, perf: 24.15, time: 1738877594.9593625} +bert-tf32-fp16: + observations: + - {batch_size: 24, cpu: 8, memory: 11938 MiB, perf: 85.36, time: 1738868844.914487} + - {batch_size: 40, cpu: 8, memory: 18916 MiB, perf: 88.91, time: 1738877678.5644176} +convnext_large-fp16: + observations: + - {batch_size: 120, cpu: 8, memory: 25022 MiB, perf: 143.83, time: 1738868043.149893} + - {batch_size: 136, cpu: 8, memory: 28270 MiB, perf: 145.07, time: 1738876850.1106644} +convnext_large-fp32: + observations: + - {batch_size: 120, cpu: 8, memory: 43770 MiB, perf: 33.28, time: 1738867967.2458732} +convnext_large-tf32: + observations: + - {batch_size: 120, cpu: 8, memory: 43770 MiB, perf: 33.19, time: 1738868211.0544605} +convnext_large-tf32-fp16: + observations: + - {batch_size: 120, cpu: 8, memory: 25022 MiB, perf: 143.86, time: 1738868293.7296622} + - {batch_size: 136, cpu: 8, memory: 28270 MiB, perf: 145.21, time: 1738876980.4151118} +diffusion-gpus: + observations: + - {batch_size: 24, cpu: 8, memory: 11688 MiB, perf: 0.0, time: 1738869732.2614098} + - {batch_size: 40, cpu: 8, memory: 11688 MiB, perf: 0.0, time: 1738878653.761756} +diffusion-single: + observations: + - {batch_size: 24, cpu: 8, memory: 22508 MiB, perf: 23.44, time: 1738869645.4673507} + - {batch_size: 40, cpu: 8, memory: 4908 MiB, perf: 0.0, time: 1738878581.8953266} +dqn: + observations: + - {batch_size: 65528, cpu: 128, memory: [358 MiB], perf: 19985985218.92, time: 1738871818.6654718} + - {batch_size: 65544, cpu: 128, memory: [358 MiB], perf: 19929722117.56, time: 1738880858.728642} +focalnet: + observations: + - {batch_size: 136, cpu: 8, memory: 23902 MiB, perf: 23.38, time: 1738878516.2305138} +lightning: + observations: + - {batch_size: 248, cpu: 8, memory: 24336 MiB, perf: 0.0, time: 1738870380.2521822} + - {batch_size: 264, cpu: 8, memory: 25906 MiB, perf: 0.0, time: 1738879315.7127686} +lightning-gpus: + observations: + - {batch_size: 248, cpu: 8, memory: 24702 MiB, perf: 0.0, time: 1738870990.8415666} + - {batch_size: 264, cpu: 8, memory: 26274 MiB, perf: 0.0, time: 1738879924.7161212} +llava-single: + observations: + - {batch_size: 1, cpu: 4, memory: 14196 MiB, perf: 0.0, time: 1738872385.6476572} + - {batch_size: 9, cpu: 4, memory: 14100 MiB, perf: 0.0, time: 1738881613.2973826} +ppo: + observations: + - {batch_size: 24, cpu: 128, memory: 716 MiB, perf: 0.0, time: 1738871891.237798} + - {batch_size: 40, cpu: 128, memory: [208 MiB], perf: 35855764.99, time: 1738881004.235976} +reformer: + observations: + - {batch_size: 24, cpu: 8, memory: 9824 MiB, perf: 17.13, time: 1738869078.5527718} + - {batch_size: 40, cpu: 8, memory: 15810 MiB, perf: 17.27, time: 1738877981.0347817} +regnet_y_128gf: + observations: + - {batch_size: 56, cpu: 8, memory: 27364 MiB, perf: 39.53, time: 1738868435.6165757} + - {batch_size: 72, cpu: 8, memory: 34570 MiB, perf: 39.25, time: 1738877152.5321622} resnet50: observations: - - {batch_size: 248, cpu: 8, memory: 12798 MiB, perf: 654.31} + - {batch_size: 248, cpu: 8, memory: 12798 MiB, perf: 646.03, time: 1738865971.4954393} + - {batch_size: 248, cpu: 8, memory: 12798 MiB, perf: 646.29, time: 1738867084.0753675} + - {batch_size: 264, cpu: 8, memory: 13614 MiB, perf: 671.57, time: 1738876039.1276622} +resnet50-noio: + observations: + - {batch_size: 248, cpu: 8, memory: 25906 MiB, perf: 636.52, time: 1738866035.8464594} + - {batch_size: 248, cpu: 8, memory: 25906 MiB, perf: 636.45, time: 1738867159.2579808} + - {batch_size: 264, cpu: 8, memory: 27538 MiB, perf: 635.71, time: 1738876106.652327} +rlhf-gpus: + observations: + - {batch_size: 56, cpu: 0, memory: 20906 MiB, perf: 1492.03, time: 1738873627.5891917} + - {batch_size: 72, cpu: 0, memory: 25544 MiB, perf: 1690.76, time: 1738882887.7195191} +rlhf-single: + observations: + - {batch_size: 56, cpu: 0, memory: 19614 MiB, perf: 459.62, time: 1738873010.649745} + - {batch_size: 72, cpu: 0, memory: 23954 MiB, perf: 0.0, time: 1738882274.9696548} +t5: + observations: + - {batch_size: 8, cpu: 8, memory: 17586 MiB, perf: 16.15, time: 1738868927.0779362} +torchatari: + observations: + - {batch_size: 1056, cpu: 34, memory: 3266 MiB, perf: 4221.24, time: 1738872323.3341098} + - {batch_size: 1072, cpu: 34, memory: 2870 MiB, perf: 4064.69, time: 1738881493.7298462} version: 2 +vjepa-gpus: + observations: + - {batch_size: 16, cpu: 12, memory: 44764 MiB, perf: 0.0, time: 1738873935.468795} + - {batch_size: 32, cpu: 12, memory: 45284 MiB, perf: 0.0, time: 1738883794.0136437} +vjepa-single: + observations: + - {batch_size: 16, cpu: 12, memory: 27292 MiB, perf: 0.0, time: 1738873773.4402642} + - {batch_size: 32, cpu: 12, memory: 40240 MiB, perf: 0.0, time: 1738883332.9009664} +whisper: + observations: + - {batch_size: 56, cpu: 8, memory: 7282 MiB, perf: 197.2, time: 1738869155.7268386} + - {batch_size: 72, cpu: 8, memory: 9126 MiB, perf: 197.67, time: 1738878065.7263389} diff --git a/milabench/sizer.py b/milabench/sizer.py index e0ef6dca..b7b0aa2d 100644 --- a/milabench/sizer.py +++ b/milabench/sizer.py @@ -3,6 +3,7 @@ import multiprocessing import os from copy import deepcopy +import time import numpy as np import yaml @@ -391,7 +392,7 @@ def on_end(self, entry): if self.rc[self.benchname] == 0: self.push_observation() else: - syslog("MemoryUsageExtractor: Could not add scaling data because of a failure") + syslog("MemoryUsageExtractor: Could not add scaling data because of a failure {}", self.benchname) self.benchname = None self.batch_size = None @@ -413,6 +414,7 @@ def push_observation(self): "batch_size": int(self.stats["batch_size"].avg), "memory": f"{int(self.stats['memory'].max)} MiB", "perf": float(f"{self.stats['perf'].avg:.2f}"), + "time": time.time() } if memorypeak := self.stats.pop("memorypeak", None):