Fix PyTorch version specification in requirements files

- Remove invalid local version labels from torch requirements
- Use proper pip syntax: torch>=2.0.0 with --index-url for CUDA/ROCm
- Fix pip installation errors for CUDA and ROCm builds
parent 97b7c6fd
Pipeline #190 canceled with stages
Flask>=2.0.0 Flask>=2.0.0
torch>=2.0.0+cu118 --index-url https://download.pytorch.org/whl/cu118 torch>=2.0.0 --index-url https://download.pytorch.org/whl/cu118
transformers>=4.30.0 transformers>=4.30.0
opencv-python>=4.5.0 opencv-python>=4.5.0
psutil>=5.8.0 psutil>=5.8.0
......
Flask>=2.0.0 Flask>=2.0.0
torch>=2.0.0+rocm5.6 --index-url https://download.pytorch.org/whl/rocm5.6 torch>=2.0.0 --index-url https://download.pytorch.org/whl/rocm5.6
transformers>=4.30.0 transformers>=4.30.0
opencv-python>=4.5.0 opencv-python>=4.5.0
psutil>=5.8.0 psutil>=5.8.0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment