Update requirements-ROCM-Linux.txt

TNTwise · Jul 5, 2024 · c657101 · c657101
1 parent c2f0fdb
commit c657101
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements-ROCM-Linux.txt b/requirements-ROCM-Linux.txt
@@ -1,9 +1,9 @@
-https://download.pytorch.org/whl/nightly/pytorch_triton_rocm-3.0.0%2B01cbe5045a-cp311-cp311-linux_x86_64.whl
+https://download.pytorch.org/whl/pytorch_triton_rocm-2.3.1-cp311-cp311-linux_x86_64.whl
 https://github.com/TNTwise/realcugan-ncnn-py/releases/download/2024-04-22-rve/realcugan_ncnn_py-2.0.0-cp311-none-manylinux1_x86_64.whl
 https://github.com/TNTwise/Universal-NCNN-upscaler-python/releases/download/2024-05-11-rve/upscale_ncnn_py-1.2.0-cp311-none-manylinux1_x86_64.whl
 https://github.com/TNTwise/rife-ncnn-vulkan-python-test/releases/download/disablesimplevk/rife_ncnn_vulkan_python-1.2.1-cp311-cp311-linux_x86_64.whl
-https://download.pytorch.org/whl/nightly/rocm6.1/torch-2.5.0.dev20240629%2Brocm6.1-cp311-cp311-linux_x86_64.whl
-https://download.pytorch.org/whl/nightly/rocm6.1/torchvision-0.20.0.dev20240629%2Brocm6.1-cp311-cp311-linux_x86_64.whl
+https://download.pytorch.org/whl/rocm5.7/torch-2.3.1%2Brocm5.7-cp311-cp311-linux_x86_64.whl
+https://download.pytorch.org/whl/rocm5.7/torchvision-0.18.1%2Brocm5.7-cp311-cp311-linux_x86_64.whl
 spandrel
 fastrlock
 onnx