If using GPU on Windows 10/11 Pro 64-bit, we recommend using Windows installers. This excludes DocTR and PlayWright support.
For newer builds of windows versions of 10/11.
-
Download Visual Studio 2022: Download Link
- Run Installer, click ok to run, click Continue
- Click on
Individual Components
- Search for these in the search bar and click on them:
- Click Install, and follow through installation, and do not need to launch VS 2022 at end.
-
Download the MinGW installer: MiniGW
-
Download and install Miniconda
-
Run Miniconda shell (not powershell!) as Administrator
-
Run:
set path=%path%;c:\MinGW\msys\1.0\bin\
to get C++ in path. In some cases it may be instead correct to useset path=%path%;c:\MinGW\bin\
-
Download latest nvidia driver for windows if one has old drivers before CUDA 11.7 supported
-
Confirm can run
nvidia-smi
and see driver version -
Setup Conda Environment:
conda create -n h2ogpt -y conda activate h2ogpt conda install python=3.10 -c conda-forge -y python --version # should say python 3.10.xx python -c "import os, sys ; print('hello world')" # should print "hello world"
-
GPU Only: Install CUDA
conda install cudatoolkit=11.7 -c conda-forge -y set CUDA_HOME=$CONDA_PREFIX
-
Install Git:
conda install -c conda-forge git
-
Install h2oGPT:
git clone https://github.com/h2oai/h2ogpt.git cd h2ogpt
-
Prepare to install dependencies:
set PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cu118"
Choose cu118+ for A100/H100+. Or for CPU set
set PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu"
-
Install primary dependencies.
- Remove any bad dependencies that existed (required for new transformers it seems):
pip uninstall flash-attn pip install -r requirements.txt
- Remove any bad dependencies that existed (required for new transformers it seems):
-
Optional: for bitsandbytes 4-bit and 8-bit:
pip uninstall bitsandbytes -y pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.1-py3-none-win_amd64.whl
-
Install document question-answer dependencies
Prefix each pip install with
--extra-index-url https://download.pytorch.org/whl/cu118
for GPU install:# Required for Doc Q/A: LangChain: pip install -r reqs_optional/requirements_optional_langchain.txt # Required for CPU: LLaMa/GPT4All: pip install -r reqs_optional/requirements_optional_gpt4all.txt # Optional: PyMuPDF/ArXiv: pip install -r reqs_optional/requirements_optional_langchain.gpllike.txt # Optional: Selenium/PlayWright: pip install -r reqs_optional/requirements_optional_langchain.urls.txt # Optional: for supporting unstructured package python -m nltk.downloader all # Optional but required for PlayWright playwright install --with-deps # Note: for Selenium, we match versions of playwright so above installer will add chrome version needed
-
AutoGPTQ support:
pip uninstall -y auto-gptq # GPU pip install auto-gptq==0.5.1 --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/ # or CPU pip install auto_gptq==0.5.1 # in-transformers support of AutoGPTQ, requires also auto-gptq above to be installed since used internally by transformers/optimum pip install optimum==1.14.1
-
AutoAWQ support:
pip uninstall -y autoawq pip install autoawq==0.1.7
-
Exllama support (GPU only):
pip uninstall -y exllama pip install https://github.com/jllllll/exllama/releases/download/0.0.18/exllama-0.0.18+cu118-cp310-cp310-win_amd64.whl --no-cache-dir
-
GPU Optional: Support LLaMa.cpp with CUDA via llama-cpp-python:
- Download/Install CUDA llama-cpp-python wheel or https://github.com/abetlen/llama-cpp-python/releases, or choose link and run pip directly. E.g.:
- GGUF ONLY for CUDA GPU (keeping CPU package in place to support CPU + GPU at same time):
pip uninstall -y llama-cpp-python-cuda pip install https://github.com/jllllll/llama-cpp-python-cuBLAS-wheels/releases/download/textgen-webui/llama_cpp_python_cuda-0.2.19+cu118-cp310-cp310-win_amd64.whl --extra-index-url https://download.pytorch.org/whl/cu118
- GGUF ONLY for CPU-AVX (can be used with -cuda one above)
For CPU, ensure to run with
pip uninstall -y llama-cpp-python pip install https://github.com/jllllll/llama-cpp-python-cuBLAS-wheels/releases/download/cpu/llama_cpp_python-0.2.19+cpuavx2-cp310-cp310-win_amd64.whl
CUDA_VISIBLE_DEVICES=
in case torch with CUDA installed.CUDA_VISIBLE_DEVICES= python generate.py --base_model=llama --prompt_type=mistral --model_path_llama=https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.1-GGUF/resolve/main/mistral-7b-instruct-v0.1.Q4_K_M.gguf --max_seq_len=4096 --score_model=None
- GGUF ONLY for CUDA GPU (keeping CPU package in place to support CPU + GPU at same time):
- If any issues, then must compile llama-cpp-python with CUDA support:
pip uninstall -y llama-cpp-python set LLAMA_CUBLAS=1 set CMAKE_ARGS=-DLLAMA_CUBLAS=on set FORCE_CMAKE=1 pip install llama-cpp-python==0.2.19 --no-cache-dir --verbose
- By default, we set
n_gpu_layers
to large value, so llama.cpp offloads all layers for maximum GPU performance. You can control this by passing--llamacpp_dict="{'n_gpu_layers':20}"
for value 20, or setting in UI. For highest performance, offload all layers. That is, one gets maximum performance if one sees in startup of h2oGPT all layers offloaded:llama_model_load_internal: offloaded 35/35 layers to GPU
but this requires sufficient GPU memory. Reduce if you have low memory GPU, say 15.
- Pass to
generate.py
the option--max_seq_len=2048
or some other number if you want model have controlled smaller context, else default (relatively large) value is used that will be slower on CPU. - If one sees
/usr/bin/nvcc
mentioned in errors, that file needs to be removed as would likely conflict with version installed for conda. - Note that once
llama-cpp-python
is compiled to support CUDA, it no longer works for CPU mode, so one would have to reinstall it without the above options to recovers CPU mode or have a separate h2oGPT env for CPU mode.
- Download/Install CUDA llama-cpp-python wheel or https://github.com/abetlen/llama-cpp-python/releases, or choose link and run pip directly. E.g.:
-
GPU Optional: Support attention sinks for infinite generation
pip install attention_sinks --no-deps
-
SERP for search:
pip install -r reqs_optional/requirements_optional_agents.txt
For more info see SERP Docs.
-
For supporting Word and Excel documents, if you don't have Word/Excel already, then download and install libreoffice: https://www.libreoffice.org/download/download-libreoffice/ .
-
To support OCR, download and install tesseract, see also: Tesseract Documentation. Please add the installation directories to your PATH.
-
vLLM support:
pip install https://h2o-release.s3.amazonaws.com/h2ogpt/openvllm-0.28.1-py3-none-any.whl
-
PDF Viewer support (only if using gradio4):
#pip install https://h2o-release.s3.amazonaws.com/h2ogpt/gradio_pdf-0.0.3-py3-none-any.whl
-
TTS and STT support (no Coqui support):
pip install pydub==0.25.1 librosa==0.10.1 ffmpeg==1.4 yt_dlp==2023.10.13 wavio==0.0.8 pip install playsound==1.3.0 pip install torchaudio soundfile==0.12.1
See FAQ for many ways to run models. The below are some other examples.
Note models are stored in C:\Users\<user>\.cache\
for chroma, huggingface, selenium, torch, weaviate, etc. directories.
-
For document Q/A with UI using LLaMa.cpp-based model on CPU or GPU:
-
Choose some GGUF model by TheBloke, then do:
python generate.py --base_model='llama' --prompt_type=llama2 --score_model=None --langchain_mode='UserData' --user_path=user_path --model_path_llama=https://huggingface.co/TheBloke/Llama-2-7b-Chat-GGUF/resolve/main/llama-2-7b-chat.Q6_K.gguf --max_seq_len=4096
Choose some other
model_path_llama
from TheBloke if desired, e.g. 13B. If no model passed, the 7B LLaMa-2 GGUF is used. For an absolute windows path, change to--user_path=C:\Users\YourUsername\h2ogpt
or something similar for some userYourUsername
. If llama-cpp-python was compiled with CUDA support, you should see in the output:Device 0: NVIDIA GeForce RTX 3090 Ti
-
Go to
http://127.0.0.1:7860
(ignore message above). Add--share=True
to get sharable secure link. -
To just chat with LLM, click
Resources
and clickLLM
in Collections, or start without--langchain_mode=UserData
. -
In
nvidia-smi
or some other GPU monitor program you should seepython.exe
using GPUs inC
(Compute) mode and using GPU resources. -
If you have multiple GPUs, best to specify to use the fasted GPU by doing (e.g. if device 0 is fastest and largest memory GPU):
set CUDA_VISIBLE_DEVICES=0
-
On an i9 with 3090Ti, one gets about 5 tokens/second.
-
For LLaMa2 70B model, launch as
python generate.py --base_model=llama --model_path_llama=https://huggingface.co/TheBloke/Llama-2-7b-Chat-GGUF/resolve/main/llama-2-7b-chat.Q6_K.gguf n_gqa=8
-
-
To use Hugging Face type models (faster on GPU than LLaMa.cpp if one has a powerful GPU with enough memory):
python generate.py --base_model=h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3 --langchain_mode=UserData --score_model=None
- On an i9 with 3090Ti, one gets about 9 tokens/second.
-
To use Hugging Face type models in 8-bit do:
python generate.py --base_model=h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3 --langchain_mode=UserData --score_model=None --load_8bit=True
When running windows on GPUs with bitsandbytes in 8-bit you should see something like the below in output:
bin C:\Users\pseud\.conda\envs\h2ogpt\lib\site-packages\bitsandbytes\libbitsandbytes_cuda117.dll
- On an i9 with 3090Ti, one gets about 5 tokens/second, so about half 16-bit speed.
- You can confirm GPU use via
nvidia-smi
showing GPU memory consumed is less than 16-bit, at about 9.2GB when in use. Also try 13B models in 8-bit for similar memory usage. - Note 8-bit inference is about twice slower than 16-bit inference, and the only use of 8-bit is to keep memory profile low.
- Bitsandbytes can be uninstalled (
pip uninstall bitsandbytes
) and still h2oGPT can be used if one does not pass--load_8bit=True
.
-
To use Hugging Face type models in 4-bit do:
python generate.py --base_model=h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3 --langchain_mode=UserData --score_model=None --load_4bit=True
- On an i9 with 3090Ti, one gets about 4 tokens/second, so still about half 16-bit speed. Memory use is about 6.6GB.
See CPU and GPU for some other general aspects about using h2oGPT on CPU or GPU, such as which models to try, quantization, etc.
- SSL Certification failure when connecting to Hugging Face.
- Your org may be blocking HF
- Try: https://stackoverflow.com/a/75111104
- Or try: huggingface/transformers#17611 (comment)
- Try using proxy.
- If you see import problems, then try setting
PYTHONPATH
in a.bat
file:for some options ...SET PYTHONPATH=.:src:$PYTHONPATH python generate.py ...
- For easier handling of command line operations, consider using bash in windows with coreutils.