From f798fd3405498125cf4a8a74f6eb07608c43ae70 Mon Sep 17 00:00:00 2001 From: Mohit Sharma Date: Thu, 23 Nov 2023 13:11:45 +0000 Subject: [PATCH] added pytorch installation step --- docs/source/onnxruntime/usage_guides/amdgpu.mdx | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/source/onnxruntime/usage_guides/amdgpu.mdx b/docs/source/onnxruntime/usage_guides/amdgpu.mdx index cde46349a30..16ee06b350a 100644 --- a/docs/source/onnxruntime/usage_guides/amdgpu.mdx +++ b/docs/source/onnxruntime/usage_guides/amdgpu.mdx @@ -16,6 +16,10 @@ To install ROCM 5.7, please follow the [ROCm installation guide](https://rocm.do #### 2. PyTorch Installation with ROCm Support Optimum ONNX Runtime integration relies on some functionalities of Transformers that require PyTorch. For now, we recommend to use Pytorch compiled against RoCm 5.7, that can be installed following [PyTorch installation guide](https://pytorch.org/get-started/locally/): +```bash +pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7 +``` + For docker installation, the following base image is recommended: `rocm/pytorch:rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1`