From bd5e0c1d56b82b0acee3383e614c0bcd047c59cb Mon Sep 17 00:00:00 2001
From: a-One-Fan <espamholding@gmail.com>
Date: Sun, 30 Apr 2023 19:13:25 +0300
Subject: [PATCH] Add initial setup file

---
 setup-oneapi.sh | 74 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100755 setup-oneapi.sh

diff --git a/setup-oneapi.sh b/setup-oneapi.sh
new file mode 100755
index 0000000..9325401
--- /dev/null
+++ b/setup-oneapi.sh
@@ -0,0 +1,74 @@
+#!/bin/bash
+
+doguide=0
+
+if [[ ! -e /opt/intel/oneapi/setvars.sh ]]; then
+    echo "You don't have the intel oneAPI base toolkit installed."
+    doguide=1
+else
+    read -p "Would you like to review the links for resources for setting up oneAPI/your drivers?" yn
+    if [[ $yn == [yY] ]]; then
+        doguide=1
+    fi
+fi
+
+if [[ $doguide == 1 ]]; then
+    echo "First, follow this guide for installing your GPU drivers, if you haven't already:"
+    echo "https://dgpu-docs.intel.com/installation-guides/ubuntu/ubuntu-jammy-arc.html"
+    echo "Afterwards, follow this guide for installing the oneAPI base toolkit:"
+    echo "https://www.intel.com/content/www/us/en/developer/tools/oneapi/base-toolkit-download.html?operatingsystem=linux&distributions=aptpackagemanager"
+fi
+
+# Python should be 3.10 specifically for now (3.9 might work but let's use 3.10)
+python3.10 --version
+if [[ $? != 0 ]]; then
+    echo "Python 3.10 is needed, and will be installed..."
+    sudo apt install python3.10
+fi
+
+# Dependency spam, most of this should be needed. libjpeg9 exists because otherwise torchvision complains.
+sudo apt-get install build-essential intel-oneapi-mkl intel-level-zero-gpu level-zero intel-opencl-icd intel-media-va-driver-non-free libmfx1 libgl-dev intel-oneapi-compiler-dpcpp-cpp libjpeg9
+
+
+# get local dependencies
+git submodule init
+git submodule update --remote
+# setup venv
+python3.10 -m venv venv
+source ./venv/bin/activate
+python3 -m pip install --upgrade pip # just to be safe
+
+# IPEX
+python -m pip install torch==1.13.0a0+git6c9b55e torchvision==0.14.1a0 intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu
+
+# Non-mandatory torchaudio install
+pip list | grep torchaudio
+if [[ $? == 1 ]]; then
+    echo "torchaudio is not installed. You may install one yourself (venv location is ./venv/bin/activate)"
+    read -p "Would you like the setup to install a custom built wheel for you, also installing/using megatools? (No exits setup) Y/N:" yn
+    if [[ $yn == [yY] ]]; then
+        sudo apt install megatools
+        mkdir temp_torchaudio_wheel
+        cd ./temp_torchaudio_wheel
+        megadl https://mega.nz/file/jc4SnCgS#_7vProAZGAw1l0vojLk7Reo7UZEUCeeHT0ye12Haow8
+        pip install --force-reinstall --no-deps torchaudio-0.13.1+b90d798-cp310-cp310-linux_x86_64.whl
+        cd ..
+        rm -rf ./temp_torchaudio_wheel
+    else
+        echo "Exiting..."
+        exit 1
+    fi
+fi
+
+# install requirements
+python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements
+python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe
+python3 -m pip install -r ./modules/dlas/requirements.txt # instal DLAS requirements
+python3 -m pip install -e ./modules/dlas/ # install DLAS
+python3 -m pip install -r ./requirements.txt # install local requirements
+
+rm *.bat
+
+sed -i 's|bitch|BEATCH\nsource /opt/intel/oneapi/setvars.sh|' ./start.sh
+
+deactivate
\ No newline at end of file