From e498b4d09ba2f4328572ee22a27f94377359a4b2 Mon Sep 17 00:00:00 2001
From: pnunna93 <104791500+pnunna93@users.noreply.github.com>
Date: Wed, 31 Jul 2024 16:25:40 -0500
Subject: [PATCH] Remove pre-release option for torch install

---
 docs/source/installation.mdx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx
index 3ed694ac1..e92981a3b 100644
--- a/docs/source/installation.mdx
+++ b/docs/source/installation.mdx
@@ -155,7 +155,7 @@ docker run -it --device=/dev/kfd --device=/dev/dri --group-add video rocm/dev-ub
 apt-get update && apt-get install -y git && cd home
 
 # Install pytorch compatible with above ROCm version
-pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.1/
+pip install torch --index-url https://download.pytorch.org/whl/nightly/rocm6.1/
 
 # Install bitsandbytes from PyPI
 # (This is supported on Ubuntu 22.04, Python 3.10, ROCm 6.1.0/6.1.1/6.1.2 and gpu arch - gfx90a, gfx942, gfx1100