update xpu requirements and image tags (#325)

Signed-off-by: Srikanth Ramakrishna <[email protected]>

Author: sramakintel

pytorch/README.md

@@ -24,7 +24,7 @@ The images below include support for both CPU and GPU optimizations:
 
 | Tag(s)                 | Pytorch  | IPEX           | Driver | Dockerfile      |
 | ---------------------- | -------- | -------------- | ------ | --------------- |
-| `2.1.40-xpu`           | [v2.1.0] | [v2.1.40+xpu]  | [914]  | [v0.4.0-Beta]   |
+| `2.1.40-xpu-pip-base`,`2.1.40-xpu`           | [v2.1.0] | [v2.1.40+xpu]  | [914]  | [v0.4.0-Beta]   |
 | `2.1.30-xpu`           | [v2.1.0] | [v2.1.30+xpu]  | [803]  | [v0.4.0-Beta]   |
 | `2.1.20-xpu`           | [v2.1.0] | [v2.1.20+xpu]  | [803]  | [v0.3.4]        |
 | `2.1.10-xpu`           | [v2.1.0] | [v2.1.10+xpu]  | [736]  | [v0.2.3]        |

pytorch/xpu-requirements.txt

@@ -6,3 +6,7 @@ oneccl_bind_pt==2.1.400+xpu
 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
 setuptools==69.5.1
 numpy==1.26.4
+idna==3.7
+requests==2.32.0
+tqdm==4.66.3
+urllib3==2.2.2

tensorflow/README.md

@@ -16,7 +16,7 @@ The images below include support for both CPU and GPU optimizations:
 
 | Tag(s)                 | TensorFlow  | ITEX           | Driver  | Dockerfile      |
 | ---------------------- | ----------- | -------------- | ------- | --------------- |
-| `2.15.0.1-xpu`, `xpu`  | [v2.15.1]   | [v2.15.0.1]    | [803.63]| [v0.4.0-Beta]   |
+| `2.15.0.1-xpu-pip-base`, `xpu`  | [v2.15.1]   | [v2.15.0.1]    | [803.63]| [v0.4.0-Beta]   |
 | `2.15.0.0-xpu`         | [v2.15.0]   | [v2.15.0.0]    | [803]   | [v0.4.0-Beta]   |
 | `2.14.0.1-xpu`         | [v2.14.1]   | [v2.14.0.1]    | [736]   | [v0.3.4]        |
 | `2.13.0.0-xpu`         | [v2.13.0]   | [v2.13.0.0]    | [647]   | [v0.2.3]        |
@@ -37,7 +37,7 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s
 
 | Tag(s)        | TensorFlow  | IPEX          | Driver | Dockerfile      |
 | ------------- | ----------- | ------------- | ------ | --------------- |
-| `2.15.0.1-xpu-jupyter` | [v2.15.1] | [v2.15.0.1]    | [803.63]| [v0.4.0-Beta]   |
+| `2.15.0.1-xpu-pip-jupyter` | [v2.15.1] | [v2.15.0.1]    | [803.63]| [v0.4.0-Beta]   |
 | `xpu-jupyter` | [v2.14.1]   | [v2.14.0.1]   | [736]  | [v0.3.4]   |
 
 ### Run the XPU Jupyter Container
@@ -49,7 +49,7 @@ docker run -it --rm \
     --device /dev/dri \
     -v /dev/dri/by-path:/dev/dri/by-path \
     --ipc=host \
-    intel/intel-extension-for-tensorflow:2.15.0.1-xpu-jupyter
+    intel/intel-extension-for-tensorflow:2.15.0.1-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.

tensorflow/xpu-requirements.txt

@@ -1,2 +1,6 @@
 tensorflow==2.15.0
 intel-extension-for-tensorflow[xpu]==2.15.0.1
+idna==3.7
+requests==2.32.0
+tqdm==4.66.3
+urllib3==2.2.2