intel / ai-containers

This repository contains Dockerfiles, scripts, yaml files, Helm charts, etc. used to scale out AI containers with versions of TensorFlow and PyTorch that have been optimized for Intel platforms. Scaling is done with python, Docker, kubernetes, kubeflow, cnvrg.io, Helm, and other container orchestration frameworks for use in the cloud and on-premise
https://intel.github.io/ai-containers/
Apache License 2.0
23 stars 18 forks source link

XPU Jupyter containers do not appear to have Jupyter installed in them #207

Closed jketreno closed 2 months ago

jketreno commented 3 months ago

Describe the bug

I was trying to use the images documented here. However, they do not appear to have Jupyter in them:

docker run -it --rm \
    -p 8888:8888 \
    --device /dev/dri \
    -v /dev/dri/by-path:/dev/dri/by-path \
    --ipc=host \
    intel/intel-extension-for-pytorch:2.1.20-xpu-pip-base
(idp) root@5711063c540f:/# python -m jupyter --version
/opt/conda/envs/idp/bin/python: No module named jupyter

Within the container, if I manually install jupyter then the server is there:

(idp) root@5711063c540f:/# cat << EOF > requirements.txt
jupyterlab==4.3.0a0
jupyterhub==5.0.0
notebook==7.3.0a0
jupyter-server-proxy>=4.1.2
EOF
(idp) root@5711063c540f:/# pip install -r requirements.txt
(idp) root@5711063c540f:/# python -m jupyter --version
Selected Jupyter core packages...
IPython          : 8.26.0
ipykernel        : 6.29.5
ipywidgets       : not installed
jupyter_client   : 8.6.2
jupyter_core     : 5.7.2
jupyter_server   : 2.14.1
jupyterlab       : 4.3.0a0
nbclient         : 0.10.0
nbconvert        : 7.16.4
nbformat         : 5.10.4
notebook         : 7.3.0a0
qtconsole        : not installed
traitlets        : 5.14.3

Error Logs

/opt/conda/envs/idp/bin/python: No module named jupyter

Reproduction Instructions

docker run -it --rm \
    -p 8888:8888\
    --device /dev/dri \
    -v /dev/dri/by-path:/dev/dri/by-path \
    --ipc=host \
    intel/intel-extension-for-pytorch:2.1.20-xpu-pip-base \
    python -m jupyter --version
/opt/conda/envs/idp/bin/python: No module named jupyter

I have tried the tags xpu-jupyter, 2.1.20-xpu-pip-base, and 2.1.10-xpu-pip-base all with the same effect.


### Affected Subfolder

- [ ] classical-ml
- [ ] enterprise
- [ ] preset
- [ ] python
- [X] pytorch
- [ ] tensorflow
- [ ] test-runner
- [ ] workflows

### Versions

```Markdown
Using the image published at intel/intel-extension-for-pytorch:{xpu-jupyter,2.1.20-xpu-pip-base,2.1.10-xpu-pip-base}
sramakintel commented 3 months ago

@jketreno thanks for reaching out. The image you are trying to pull and test does not have jupyter in it. Please you this command/image: docker pull intel/intel-extension-for-pytorch:2.1.20-xpu-pip-jupyter

jketreno commented 3 months ago

Thanks. Can you update https://github.com/intel/ai-containers/blob/main/pytorch/README.md#xpu-images to list/use the correct tags?

Thanks.

image