Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix PyTorch XPU Jupyter Documentation #224

Merged
merged 11 commits into from
Jul 8, 2024
Merged

Fix PyTorch XPU Jupyter Documentation #224

Changes from 5 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
916d868
correct documentation
sramakintel Jul 8, 2024
4abcc30
correct documentation in docker run
sramakintel Jul 8, 2024
1bab66f
separate pip and idp sections
sramakintel Jul 8, 2024
940e58b
correct version
sramakintel Jul 8, 2024
f9efbd9
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
sramakintel Jul 8, 2024
d6bfde1
correct lint errors
sramakintel Jul 8, 2024
25e99b3
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
sramakintel Jul 8, 2024
56fa1a0
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
sramakintel Jul 8, 2024
5d40546
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
sramakintel Jul 8, 2024
8a4f088
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
sramakintel Jul 8, 2024
9ba27ba
Merge branch 'main' into sramakr1/update_docs_xpu_jupyter
Jul 8, 2024
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
23 changes: 19 additions & 4 deletions pytorch/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,9 +45,8 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s

| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
| `xpu-jupyter` | [v2.1.0] | [v2.1.30+xpu] | [803] | `8888` | [v0.4.0-Beta] |
| `2.1.20-xpu-pip-base` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
| `2.1.10-xpu-pip-base` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.3.4] |
| `2.1.20-xpu-pip-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
| `2.1.10-xpu-pip-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |

### Run the XPU Jupyter Container

Expand All @@ -58,7 +57,7 @@ docker run -it --rm \
--device /dev/dri \
-v /dev/dri/by-path:/dev/dri/by-path \
--ipc=host \
intel/intel-extension-for-pytorch:xpu-jupyter
intel/intel-extension-for-pytorch:2.1.20-xpu-pip-jupyter
```

After running the command above, copy the URL (something like `http://127.0.0.1:$PORT/?token=***`) into your browser to access the notebook server.
Expand Down Expand Up @@ -246,6 +245,22 @@ The images below additionally include [Intel® oneAPI Collective Communications
| `2.1.0-idp-mulitnode` | [v2.1.0] | [v2.1.0+cpu] | [v2.1.0][ccl-v2.1.0] | [v2.3.1] | [v0.2.3] |
| `2.0.0-idp-multinode` | [v2.0.0] | [v2.0.0+cpu] | [v2.0.0][ccl-v2.0.0] | [v2.1.1] | [v0.1.0] |

## XPU images with Intel® Distribution for Python*

The images below are built only with CPU and GPU optimizations and include [Intel® Distribution for Python*]:

| Tag(s) | Pytorch | IPEX | Dockerfile |
| ---------------- | -------- | ------------ | --------------- |
| `2.1.10-xpu-idp-base` | [v2.1.0] | [v2.1.10+xpu] | [736] | [v0.2.3] |

The images below additionally include [Jupyter Notebook](https://jupyter.org/) server:

| Tag(s) | Pytorch | IPEX | Driver | Jupyter Port | Dockerfile |
| --------------------- | -------- | ------------- | ------ | ------------ | --------------- |
| `2.1.20-xpu-idp-jupyter` | [v2.1.0] | [v2.1.20+xpu] | [803] | `8888` | [v0.3.4] |
| `2.1.10-xpu-idp-jupyter` | [v2.1.0] | [v2.1.10+xpu] | [736] | `8888` | [v0.2.3] |


## Build from Source

To build the images from source, clone the [Intel® AI Containers](https://github.com/intel/ai-containers) repository, follow the main `README.md` file to setup your environment, and run the following command:
Expand Down