Could not find a version that satisfies the requirement torch==1.11.0+cu113

[cameronk]

cog version 0.4.4 (built 2022-09-23T19:52:35Z)

Problem

Looking to set up pyannote-audio with a specific version of torch.

build:
  gpu: true
  python_version: "3.8"
  python_packages:
    - "pyannote.audio==2.1.1"
    - "torch==1.11.0"
    - "torchaudio==0.11.0"
    - "torchvision==0.12.0"
  system_packages:
    - "libsndfile1"
predict: "predict.py:Predictor"

 > [stage-0 8/9] RUN --mount=type=cache,target=/root/.cache/pip pip install   pyannote.audio==2.1.1 torch==1.11.0+cu113 torchaudio==0.11.0 torchvision==0.12.0+cu113:
#14 0.922 Collecting pyannote.audio==2.1.1
#14 0.926   Using cached pyannote.audio-2.1.1-py2.py3-none-any.whl (390 kB)
#14 1.011 ERROR: Could not find a version that satisfies the requirement torch==1.11.0+cu113 (from versions: 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0)
#14 1.016 WARNING: You are using pip version 22.0.4; however, version 22.3.1 is available.
#14 1.016 You should consider upgrading via the '/root/.pyenv/versions/3.8.15/bin/python3.8 -m pip install --upgrade pip' command.

I believe the warning at the end may be relevant, it looks it may have been recently fixed pypa/pipenv#5022

[gradjitta]

Not sure how to fix it but looks like cog is not looking at list of wheels listed in https://download.pytorch.org/whl/torch_stable.html , as those versions do exist

https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl https://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl https://download.pytorch.org/whl/cu113/torchvision-0.12.0%2Bcu113-cp38-cp38-linux_x86_64.whl

If you notice https://github.com/replicate/cog/blob/main/pkg/config/torch_compatability_matrix.json
{ "Torch": "1.11.0+cu113", "Torchvision": "0.12.0+cu113", "Torchaudio": "0.11.0", "IndexURL": "", "CUDA": "11.3", "Pythons": [ "3.6", "3.7", "3.8", "3.9", "3.10" ] }

Maybe setting IndexURL to https://download.pytorch.org/whl/torch_stable.html is the solution?

[bfirsh]

This may be fixed by #812...

[bfirsh]

Looks like it's correctly in the matrix now so this will indeed be fixed.

[cameronk]

Awesome, thanks @bfirsh! I'll close this and keep an eye on #812.