added pytorch installation step

huggingface · Nov 23, 2023 · f798fd3 · f798fd3
1 parent e02f6ea
commit f798fd3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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
+```
+
 <Tip>
 For docker installation, the following base image is recommended: `rocm/pytorch:rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1`
 </Tip>