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 && \ ...@@ -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
......
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