Torchtune nightly can't be found in ROCm nightly wheel #1771
Description
TL;DR: The following command fails:
pip3 install --extra-index-url "https://download.pytorch.org/whl/nightly/rocm6.2" torchtune==0.3.0.dev20240928
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/nightly/rocm6.2
ERROR: Could not find a version that satisfies the requirement torchtune==0.3.0.dev20240928 (from versions: 0.0.1, 0.1.0, 0.1.1, 0.2.0, 0.2.1, 0.3.0, 0.3.1)
ERROR: No matching distribution found for torchtune==0.3.0.dev20240928
Given the other versions are available, can we add the nightly as well?
Hi all! I recently landed pytorch/torchchat#1279 in Torchchat to change the TORCH_NIGHTLY_URL to the ROCm wheel for AMD GPU systems. However, our installation script fails with the following message:
From the torchchat repo in a fresh .venv
./install/install_requirements.sh
...
+ pip3 install --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2 torch==2.5.0.dev20240901 torchvision==0.20.0.dev20240901 torchtune==0.3.0.dev20240928
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/nightly/rocm6.2
Collecting torch==2.5.0.dev20240901
Using cached https://download.pytorch.org/whl/nightly/rocm6.2/torch-2.5.0.dev20240901%2Brocm6.2-cp310-cp310-linux_x86_64.whl (3546.0 MB)
Collecting torchvision==0.20.0.dev20240901
Using cached https://download.pytorch.org/whl/nightly/rocm6.2/torchvision-0.20.0.dev20240901%2Brocm6.2-cp310-cp310-linux_x86_64.whl (2.7 MB)
ERROR: Could not find a version that satisfies the requirement torchtune==0.3.0.dev20240928 (from versions: 0.0.1, 0.1.0, 0.1.1, 0.2.0, 0.2.1, 0.3.0, 0.3.1)
ERROR: No matching distribution found for torchtune==0.3.0.dev20240928
It seems like the nightly version of Torchtune (in this case 0.3.0dev20240928) can't be found through the ROCm 6.2 wheel. Is there a way we can publish nightlies here?
Thanks!