https://download.pytorch.org/whl/ is quite useful for automation like CI. Another feature that I really like is the platform filter, example:
-
https://download.pytorch.org/whl/${arch}
, arch = cpu, cu116, etc.
However, I noticed that there is no such filter for ROCm. None of the following work, and trying these gives unauthorized access XML as return.
- https://download.pytorch.org/whl//rocm/
- prefix/rocm511/ (using prefix since new users are limited at 2 links only. prefix =
https://download.pytorch.org/whl/
) - prefix/rocm5.1.1/
Any idea what I can use for filtering wheels for ROCm?