diff --git a/docs/build/eps.md b/docs/build/eps.md index b6138ecab1f2a..0535969441b14 100644 --- a/docs/build/eps.md +++ b/docs/build/eps.md @@ -603,9 +603,11 @@ See more information on the ROCm Execution Provider [here](../execution-provider #### Linux ```bash -./build.sh --config --use_rocm --rocm_home +./build.sh --config --use_rocm --rocm_home --build_wheel --update --build --parallel ``` +This produces the .whl file in `./build/Linux//dist`. + Dockerfile instructions are available [here](https://github.com/microsoft/onnxruntime/tree/main/dockerfiles#rocm). ## NNAPI diff --git a/docs/build/training.md b/docs/build/training.md index 1840a35941220..9081f0f6a8f1c 100644 --- a/docs/build/training.md +++ b/docs/build/training.md @@ -145,13 +145,13 @@ The default AMD GPU build requires ROCm software toolkit installed on the system 2. Create the ONNX Runtime Python wheel ```bash - ./build.sh --config Release --enable_training --build_wheel --parallel --skip_tests --use_rocm --rocm_home /opt/rocm + ./build.sh --config --enable_training --build_wheel --parallel --skip_tests --use_rocm --rocm_home /opt/rocm ``` -3. Install the .whl file in `./build/Linux/RelWithDebInfo/dist` for ONNX Runtime Training. +3. Install the .whl file in `./build/Linux//dist` for ONNX Runtime Training. ```bash - python -m pip install build/Linux/RelWithDebInfo/dist/*.whl + python -m pip install build/Linux//dist/*.whl ``` ## DNNL and MKLML diff --git a/src/routes/getting-started/table.svelte b/src/routes/getting-started/table.svelte index 8bbed8f0a34cd..8608f55038e11 100644 --- a/src/routes/getting-started/table.svelte +++ b/src/routes/getting-started/table.svelte @@ -502,22 +502,22 @@ "Follow build instructions from here", 'linux,Python,X64,MIGraphX': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,C-API,X64,MIGraphX': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,C++,X64,MIGraphX': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,Python,X64,ROCm': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,C-API,X64,ROCm': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,C++,X64,ROCm': - "Follow build instructions from here", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
Build from source by following build instructions from here", 'linux,Python,ARM64,ACL': "Follow build instructions from here", @@ -691,10 +691,10 @@ 'python -m pip install cerberus flatbuffers h5py numpy>=1.16.6 onnx packaging protobuf sympy setuptools>=41.4.0
pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-training
pip install torch-ort
python -m torch_ort.configure', 'ot_linux,ot_large_model,ot_python,ot_X64,ot_ROCm,ot_stable': - "pip install onnxruntime-training -f https://download.onnxruntime.ai/onnxruntime_stable_<rocm_version*>.html
pip install torch-ort
python -m torch_ort.configure

*Available versions", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-rocm
pip install torch-ort
python -m torch_ort.configure

*Available versions", 'ot_linux,ot_large_model,ot_python,ot_X64,ot_ROCm,ot_nightly': - "pip install onnxruntime-training -f https://download.onnxruntime.ai/onnxruntime_nightly_<rocm_version*>.html
pip install torch-ort
python -m torch_ort.configure

*Available versions", + "pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-training-rocm
pip install torch-ort
python -m torch_ort.configure

*Available versions", 'ot_linux,ot_on_device,ot_python,ot_X64,ot_CPU,ot_stable': 'python -m pip install cerberus flatbuffers h5py numpy>=1.16.6 onnx packaging protobuf sympy setuptools>=41.4.0
pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-cpu',