Skip to content

Commit

Permalink
Remove pre-release option for torch install
Browse files Browse the repository at this point in the history
  • Loading branch information
pnunna93 authored Jul 31, 2024
1 parent db1df72 commit e498b4d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/installation.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e498b4d

Please sign in to comment.