Skip to content

Commit a78a7da

Browse files
committed
update torch install
Signed-off-by: chensuyue <suyue.chen@intel.com>
1 parent 74e9381 commit a78a7da

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,10 @@ support AMD CPU, ARM CPU, and NVidia GPU through ONNX Runtime with limited testi
3636
```Shell
3737
pip install torch --index-url https://download.pytorch.org/whl/cpu
3838
```
39-
#### Install torch/intel_extension_for_pytorch for XPU
39+
#### Use Docker Image with torch installed for HPU
40+
https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html#bare-metal-fresh-os-single-click
41+
42+
#### Install torch/intel_extension_for_pytorch for Intel GPU
4043
https://intel.github.io/intel-extension-for-pytorch/index.html#installation
4144

4245
#### Install torch for other platform

docs/source/installation_guide.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,10 @@ The following prerequisites and requirements must be satisfied for a successful
3232
```Shell
3333
pip install torch --index-url https://download.pytorch.org/whl/cpu
3434
```
35-
#### Install torch for XPU
35+
#### Use Docker Image with torch installed for HPU
36+
https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html#bare-metal-fresh-os-single-click
37+
38+
#### Install torch/intel_extension_for_pytorch for Intel GPU
3639
https://intel.github.io/intel-extension-for-pytorch/index.html#installation
3740

3841
#### Install torch for other platform

0 commit comments

Comments
 (0)