diff --git a/requirements_linux_ipex.txt b/requirements_linux_ipex.txt index f794a9046..752380b4b 100644 --- a/requirements_linux_ipex.txt +++ b/requirements_linux_ipex.txt @@ -1,5 +1,5 @@ -torch==2.1.0.post0+cxx11.abi torchvision==0.16.0.post0+cxx11.abi intel-extension-for-pytorch==2.1.20+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ +torch==2.1.0.post2+cxx11.abi torchvision==0.16.0.post2+cxx11.abi intel-extension-for-pytorch==2.1.30+xpu oneccl_bind_pt==2.1.300+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ tensorboard==2.15.2 tensorflow==2.15.0 intel-extension-for-tensorflow[xpu]==2.15.0.0 -mkl==2024.1.0 mkl-dpcpp==2024.1.0 oneccl-devel==2021.12.0 impi-devel==2021.12.0 +mkl==2024.1.0 mkl-dpcpp==2024.1.0 oneccl-devel==2021.12.0 impi-devel==2021.12.1 onnxruntime-openvino==1.17.1 -r requirements.txt diff --git a/requirements_linux_rocm.txt b/requirements_linux_rocm.txt index 916806471..570ace0a2 100644 --- a/requirements_linux_rocm.txt +++ b/requirements_linux_rocm.txt @@ -1,4 +1,4 @@ -torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.0 +torch==2.3.0+rocm6.0 torchvision==0.18.0+rocm6.0 --index-url https://download.pytorch.org/whl/rocm6.0 tensorboard==2.14.1 tensorflow-rocm==2.14.0.600 onnxruntime-training --pre --index-url https://pypi.lsh.sh/60/ --extra-index-url https://pypi.org/simple -r requirements.txt