Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 13 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -97,22 +97,31 @@ Includes [multiple gpus quantization](https://github.com/intel/auto-round/blob/m
### Install from pypi

```bash
# CPU/Intel GPU/CUDA
# CPU(Xeon)/GPU(CUDA)
pip install auto-round

# HPU
# HPU(Gaudi)
# install inside the hpu docker container, e.g. vault.habana.ai/gaudi-docker/1.23.0/ubuntu24.04/habanalabs/pytorch-installer-2.9.0:latest
pip install auto-round-hpu

# XPU(Intel GPU)
pip install torch --index-url https://download.pytorch.org/whl/xpu
pip install auto-round
```

<details>
<summary>Build from Source</summary>

```bash
# CPU/Intel GPU/CUDA
# CPU(Xeon)/GPU(CUDA)
pip install .

# HPU
# HPU(Gaudi)
python setup.py install hpu

# XPU(Intel GPU)
pip install torch --index-url https://download.pytorch.org/whl/xpu
pip install .
```

</details>
Expand Down
17 changes: 13 additions & 4 deletions README_CN.md
Original file line number Diff line number Diff line change
Expand Up @@ -86,22 +86,31 @@ AutoRound 是专为大语言模型(LLMs)和视觉-语言模型(VLMs)设
### 从 PyPI 安装

```shell
# CPU / Intel GPU / CUDA
# CPU(Xeon)/GPU(CUDA)
pip install auto-round

# HPU
# HPU(Gaudi)
# 在 hpu docker container 中安装, e.g. vault.habana.ai/gaudi-docker/1.23.0/ubuntu24.04/habanalabs/pytorch-installer-2.9.0:latest
pip install auto-round-hpu

# XPU(Intel GPU)
pip install torch --index-url https://download.pytorch.org/whl/xpu
pip install auto-round
```

<details>
<summary>从源码编译安装</summary>

```bash
# CPU/Intel GPU/CUDA
# CPU(Xeon)/GPU(CUDA)
pip install .

# HPU
# HPU(Gaudi)
python setup.py install hpu

# XPU(Intel GPU)
pip install torch --index-url https://download.pytorch.org/whl/xpu
pip install .
```

</details>
Expand Down
1 change: 0 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,6 @@ def fetch_requirements(path):

package_name = "auto-round"

# From v0.9.3, auto-round-hpu will be published to replace auto-round-lib.
hpu_build = "hpu" in sys.argv
if hpu_build:
sys.argv.remove("hpu")
Expand Down