Add ROCm torch-ort installation instructions (#8347)

This commit is contained in:
Nat Kershaw (MSFT) 2021-07-09 17:50:30 -07:00 committed by GitHub
parent fce58d9e26
commit ec29d22e7a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -545,7 +545,7 @@ var ot_validCombos = {
"Follow sample notebook from <a href='https://github.com/microsoft/onnxruntime-training-examples' target='_blank'>here</a>",
"ot_linux,ot_PyTorch18,ot_X64,ot_AMD":
"This combination is in preview. Please <a href=https://github.com/Microsoft/onnxruntime/issues>reach out</a> for preview packages",
"This combination requires an AMD docker image. See <a href=https://github.com/pytorch/ort>the torch-ort repo</a> for details",
"ot_linux,ot_PyTorch18,ot_X64,ot_DNNL":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/how-to/build/training' target='_blank'>build from source</a>.",
@ -557,7 +557,7 @@ var ot_validCombos = {
"pip install torch-ort -f https://onnxruntimepackages.z14.web.core.windows.net/onnxruntime_stable_torch190.cu111.html<br/> python -m torch_ort.configure",
"ot_linux,ot_PyTorch19,ot_X64,ot_AMD":
"This combination is in preview. Please <a href=https://github.com/Microsoft/onnxruntime/issues>reach out</a> for preview packages",
"(Preview) pip install torch -f https://download.pytorch.org/whl/rocm4.2/torch_stable.html<br/>pip install torch-ort -f https://onnxruntimepackages.z14.web.core.windows.net/onnxruntime_stable_torch190.rocm42.html<br/>python -m torch_ort.configure</br>",
"ot_linux,ot_C++,ot_X64,ot_CUDA10":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/how-to/build/training' target='_blank'>build from source</a>.",