UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update PyTorch wheel source to CUDA 11.8

Browse Source 
Modified the Dockerfile to install PyTorch, torchvision, and torchaudio from a CUDA 11.8 specific wheel URL. This ensures compatibility with the CUDA version in our environment and potentially improves performance and stability for GPU-accelerated operations.
This commit is contained in:
Joseph Young 2024-03-20 18:33:34 -04:00
parent 8ce48dc7d1
commit 2588da0e27
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile-cuda
View File

@ -86,7 +86,7 @@ RUN apk update && \
apk del /var/cache/apk/*.tbz2 apk del /var/cache/apk/*.tbz2
COPY ./backend/requirements.txt ./requirements.txt COPY ./backend/requirements.txt ./requirements.txt
RUN pip3 install torch torchvision torchaudio --no-cache-dir && \ RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --no-cache-dir && \
pip3 install -r requirements.txt --no-cache-dir pip3 install -r requirements.txt --no-cache-dir
# copy built frontend files # copy built frontend files