pytorch: remove unused requirements

Author: jondea

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -88,13 +88,8 @@ COPY $TORCH_AO_WHEEL /home/$DOCKER_USER/
 # will need to manually add their dependencies. We don't use the nightly
 # versions which corresponding to our torch build because they can disappear,
 # and we usually don't need features from the nightlies.
-# Note: torchvision is currently pinned to a nightly build, this can be updated
-# at the next vision release, and the `--extra-index-url` removed.
 RUN pip install \
-    torchaudio~=2.6.0 \
-    torchdata~=0.11.0 \
     torchvision~=0.23.0 \
-    torchtune~=0.5.0 \
     --no-deps
 
 # We need --no-deps because the torch version won't match the versions on torch*

ML-Frameworks/pytorch-aarch64/requirements.txt

@@ -3,7 +3,6 @@
 # PyTorch channel and our own built torch wheel (see Dockerfile for more info   )
 certifi==2024.8.30
 charset-normalizer~=3.4.0
-datasets~=3.4.1
 expecttest==0.3.0 # From unit tests
 filelock~=3.16.1
 fsspec==2024.9.0
@@ -14,13 +13,11 @@ MarkupSafe~=3.0.1
 mpmath~=1.3.0
 networkx~=3.4.1
 numpy~=2.1.2
-omegaconf~=2.3.0
 opencv-python-headless~=4.10.0.84
 packaging~=24.1
 pandas~=2.2.3
 pillow~=11.0.0
 protobuf==5.29.5      # GenAI models dependency
-psutil~=7.0.0
 pyaml~=24.9.0
 python-dateutil~=2.9.0.post0
 pytz==2024.2
@@ -30,7 +27,6 @@ requests~=2.32.3
 safetensors~=0.4.5
 six~=1.16.0
 sympy~=1.13.1
-tiktoken~=0.9.0
 tokenizers~=0.21.0
 tqdm~=4.66.5
 transformers~=4.55.2  # >= 4.50.0 due to CVE-2025-2099