"backend/apps/webui/routers/auths.py" did not exist on "298e6848b383bf5b3d4590ffa3a95f8dc8450f7b"
Commit 2588da0e authored by Joseph Young's avatar Joseph Young
Browse files

Update PyTorch wheel source to CUDA 11.8

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.
parent 8ce48dc7
......@@ -86,7 +86,7 @@ RUN apk update && \
apk del /var/cache/apk/*.tbz2
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
# copy built frontend files
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment