Skip to content

Commit

Permalink
update 3x torch installation
Browse files Browse the repository at this point in the history
Signed-off-by: chensuyue <[email protected]>
  • Loading branch information
chensuyue committed Jul 28, 2024
1 parent 6e1b1da commit 9eef63b
Show file tree
Hide file tree
Showing 3 changed files with 32 additions and 2 deletions.
17 changes: 16 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,26 @@ support AMD CPU, ARM CPU, and NVidia GPU through ONNX Runtime with limited testi
* [2024/07] Performance optimizations and usability improvements on [client-side](https://github.com/intel/neural-compressor/blob/master/docs/source/3x/client_quant.md).

## Installation
### Install Framework
#### Install torch for CPU
```Shell
pip install torch --index-url https://download.pytorch.org/whl/cpu
```
#### Install torch for XPU
https://intel.github.io/intel-extension-for-pytorch/index.html#installation

#### Install torch for other platform
https://pytorch.org/get-started/locally

#### Install tensorflow
```Shell
pip install tensorflow
```

### Install from pypi
```Shell
# Install 2.X API + Framework extension API + PyTorch dependency
pip install neural-compressor[pt]
pip install neural-compressor[pt]
# Install 2.X API + Framework extension API + TensorFlow dependency
pip install neural-compressor[tf]
```
Expand Down
16 changes: 16 additions & 0 deletions docs/source/installation_guide.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,22 @@ The following prerequisites and requirements must be satisfied for a successful
> Notes:
> - If you get some build issues, please check [frequently asked questions](faq.md) at first.
### Install Framework
#### Install torch for CPU
```Shell
pip install torch --index-url https://download.pytorch.org/whl/cpu
```
#### Install torch for XPU
https://intel.github.io/intel-extension-for-pytorch/index.html#installation

#### Install torch for other platform
https://pytorch.org/get-started/locally

#### Install tensorflow
```Shell
pip install tensorflow
```

### Install from Binary
- Install from Pypi
```Shell
Expand Down
1 change: 0 additions & 1 deletion requirements_pt.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,3 @@ prettytable
psutil
py-cpuinfo
pydantic
torch

0 comments on commit 9eef63b

Please sign in to comment.