UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-22 08:07:55 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Adding the missing env variable

Browse Source 
ENV RAG_EMBEDDING_MODEL_DEVICE_TYPE="cuda"
This commit is contained in:
Joseph Young 2024-03-20 18:16:11 -04:00
parent 9a8a48b879
commit 9bea40bd40
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile-cuda
View File

@ -32,6 +32,7 @@ RUN npm run build
######## CUDA WebUI backend ########
FROM --platform=linux/amd64 nvidia/cuda:"$CUDA_VERSION"-devel-ubuntu22.04 AS cuda-build-amd64
#FROM --platform=linux/amd64 cgr.dev/chainguard/pytorch-cuda12:latest AS cuda-build-amd64 # fails with python requirements conflicts
ENV RAG_EMBEDDING_MODEL_DEVICE_TYPE="cuda"
# Set environment variables for NVIDIA Container Toolkit
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 \