torch==2.3.0+rocm6.0 torchvision==0.18.0+rocm6.0 --index-url https://download.pytorch.org/whl/rocm6.0 | |
tensorboard==2.14.1 tensorflow-rocm==2.14.0.600 | |
onnxruntime-training --pre --index-url https://pypi.lsh.sh/60/ --extra-index-url https://pypi.org/simple | |
-r requirements.txt | |