Update recommended CMAKE flag to enable using CUDA on linux in Docs

This commit is contained in:
Debanjum Singh Solanky 2024-04-13 22:11:10 +05:30
parent 148923c13a
commit 689202e00e

View file

@ -134,7 +134,7 @@ python -m pip install khoj-assistant
# CPU
python -m pip install khoj-assistant
# NVIDIA (CUDA) GPU
CMAKE_ARGS="DLLAMA_CUBLAS=on" FORCE_CMAKE=1 python -m pip install khoj-assistant
CMAKE_ARGS="DLLAMA_CUDA=on" FORCE_CMAKE=1 python -m pip install khoj-assistant
# AMD (ROCm) GPU
CMAKE_ARGS="-DLLAMA_HIPBLAS=on" FORCE_CMAKE=1 python -m pip install khoj-assistant
# VULCAN GPU