update and test serving

Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com>

Author: sramakintel

pytorch/Dockerfile

@@ -169,7 +169,7 @@ RUN apt-get update && \
 
 RUN rm -rf /etc/apt/sources.list.d/intel-gpu-jammy.list
 
-ENV LD_LIBRARY_PATH=/opt/intel/oneapi/redist/lib:$LD_LIBRARY_PATH
+ENV OCL_ICD_VENDORS=/etc/OpenCL/vendors
 
 FROM ipex-xpu-base AS ipex-xpu-base-wheels-pip
 

pytorch/serving/model-archive/ipex_squeezenet.py

@@ -27,8 +27,8 @@
 
 # pylint: skip-file
 
-import intel_extension_for_pytorch as ipex
 import torch
+import intel_extension_for_pytorch as ipex
 import torchvision.models as models
 
 # load the model

pytorch/serving/torchserve-xpu-requirements.txt

@@ -1,9 +1,8 @@
-torch==2.1.0.post3+cxx11.abi
-torchvision==0.16.0.post3+cxx11.abi
-torchaudio==2.1.0.post3+cxx11.abi
-intel_extension_for_pytorch==2.1.40+xpu
+torch==2.3.1+cxx11.abi
+torchvision==0.18.1+cxx11.abi
+torchaudio==2.3.1+cxx11.abi
+intel_extension_for_pytorch==2.3.110+xpu
 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
-setuptools==69.5.1
 numpy==1.26.4
 captum>=0.7.0
 cython>=3.0.10