CPU install routine

This commit is contained in:
marius 2024-08-11 08:58:20 +02:00
parent 432fb22b6d
commit 9f7f0028e3
2 changed files with 42 additions and 0 deletions

13
dependencies/install_cpu.sh vendored Normal file
View File

@ -0,0 +1,13 @@
#!/usr/bin/env bash
if [[ "$(uname)" == "Darwin" ]] || [[ "$(uname)" == "Linux" ]]; then
echo "The CPU installation is starting (macOS, Linux)."
echo "Installing dependencies"
pip install -r "https://raw.githubusercontent.com/norandom/log2ml/main/dependencies/requirements.cgpu.txt"
echo "cuML will not be installed."
else
echo "This does not appear to be a CPU environment."
fi

29
dependencies/requirements.cpu.txt vendored Normal file
View File

@ -0,0 +1,29 @@
openai==1.13.3
langchain==0.1.10
python-dotenv==1.0.1
tiktoken==0.6.0
langchain_openai==0.0.8
langchain_experimental==0.0.53
langchainhub==0.1.14
ipywidgets
transformers==4.39.0
torch==2.2.1+cpu
torchvision==0.17.1+cpu
torchaudio==2.2.1+cpu
sentence-transformers==2.5.1
faiss_cpu==1.8.0
linformer-pytorch==0.19.3
PyGithub==2.3.0
deap==1.4.1
update_checker==0.18.0
scikit-mdr==0.4.5
skrebate==0.62
xgboost==2.0.3
stopit==1.1.2
tpot==0.12.2
umap-learn==0.5.6
tabulate==0.9.0
onnx==1.16.1
skl2onnx==1.17.0
protobuf==3.20.3
dill==0.3.8