# Wheels generated by http://rocm-ci.amd.com/job/pytorch-pipeline-manylinux-wheel-builder_rel-6.4/495/

wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.3/apex-1.5.0+rocm6.4.3.git118d7139-cp39-cp39-linux_x86_64.whl -O apex-1.5.0+rocm6.4.3.git118d7139-cp39-cp39-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.3/pytorch_triton_rocm-3.1.0+rocm6.4.3.git0deb5489-cp39-cp39-linux_x86_64.whl -O pytorch_triton_rocm-3.1.0+rocm6.4.3.git0deb5489-cp39-cp39-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.3/torch-2.5.1+rocm6.4.3.git618a5512-cp39-cp39-linux_x86_64.whl -O torch-2.5.1+rocm6.4.3.git618a5512-cp39-cp39-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.3/torchaudio-2.5.0+rocm6.4.3.git56bc006d-cp39-cp39-linux_x86_64.whl -O torchaudio-2.5.0+rocm6.4.3.git56bc006d-cp39-cp39-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.3/torchvision-0.20.1+rocm6.4.3.git04d8fc4a-cp39-cp39-linux_x86_64.whl -O torchvision-0.20.1+rocm6.4.3.git04d8fc4a-cp39-cp39-linux_x86_64.whl

pip3 install \
apex-1.5.0+rocm6.4.3.git118d7139-cp39-cp39-linux_x86_64.whl \
pytorch_triton_rocm-3.1.0+rocm6.4.3.git0deb5489-cp39-cp39-linux_x86_64.whl \
torch-2.5.1+rocm6.4.3.git618a5512-cp39-cp39-linux_x86_64.whl \
torchaudio-2.5.0+rocm6.4.3.git56bc006d-cp39-cp39-linux_x86_64.whl \
torchvision-0.20.1+rocm6.4.3.git04d8fc4a-cp39-cp39-linux_x86_64.whl \