Torch wheel resolution broken for torch==1.11.0 and cuda 117

[H4dr1en]

Hi,

In clearml-agent==1.5.2, if I specify in the requirements: torch==1.11.0 and the agent recognises cuda 117 (agent.cuda_version = 117 is printed in the logs),
The agent will download the cpu version of torch 1.11.0 instead of trying to downgrade the cuda version and find the matching wheel (eg. 1.11.0+cuda117 -> 1.11.0+cuda116 -> 1.11.0+cuda115 exists)

[jkhenning]

Hi @H4dr1en, can you attach the logs please?

[jkhenning]

@H4dr1en updating from the Slack discussion:
See at the attached images
https://download.pytorch.org/whl/cu117/torch/ does not contain a specific wheel for cuda117 to x86, they use the pip default one

[H4dr1en]

More on the Slack discussion > https://clearml.slack.com/archives/CTK20V944/p1683734024785499

[H4dr1en]

Seems to be fixed in https://github.com/allegroai/clearml-agent/releases/tag/v1.6.0
Not tested but closing for now