tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

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
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