1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

IPEX update to Torch 2.1 and bundle in MKL & DPCPP

pull/1772/head
Disty0 2023-12-14 12:37:22 +03:00
parent 9f2637a133
commit 7cc3045e4a
2 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gui.sh
View File

@ -72,14 +72,8 @@ fi
#Set OneAPI if it's not set by the user
if [[ "$@" == *"--use-ipex"* ]]
then
echo "Setting OneAPI environment"
if [ ! -x "$(command -v sycl-ls)" ]
then
if [[ -z "$ONEAPI_ROOT" ]]
then
ONEAPI_ROOT=/opt/intel/oneapi
fi
source $ONEAPI_ROOT/setvars.sh
if [ -d "$SCRIPT_DIR/venv" ]; then
export LD_LIBRARY_PATH=$(realpath "$SCRIPT_DIR/venv")/lib/:$LD_LIBRARY_PATH
fi
export NEOReadDebugKeys=1
export ClDeviceGlobalMemSizeAvailablePercent=100

5
requirements_linux_ipex.txt
View File

@ -1,3 +1,4 @@
torch==2.0.1a0+cxx11.abi torchvision==0.15.2a0+cxx11.abi intel_extension_for_pytorch==2.0.110+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
tensorboard==2.14.1 tensorflow==2.14.0 intel-extension-for-tensorflow[gpu]
torch==2.1.0a0+cxx11.abi torchvision==0.16.0a0+cxx11.abi intel-extension-for-pytorch==2.1.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
tensorboard==2.14.1 tensorflow==2.14.0 intel-extension-for-tensorflow[xpu]==2.14.0.1
mkl==2024.0.0 mkl-dpcpp==2024.0.0
-r requirements.txt