docker: install pytorch 11.8

This commit is contained in:
Cyberes 2023-10-14 20:41:40 -06:00
parent 076501a784
commit 448ebcbe8d
1 changed files with 2 additions and 0 deletions

View File

@ -18,6 +18,8 @@ RUN rm /tmp/rathole.zip
RUN cp /tmp/rathole /app RUN cp /tmp/rathole /app
RUN python3 -m venv /venv RUN python3 -m venv /venv
RUN /venv/bin/pip install torch --index-url https://download.pytorch.org/whl/cu118
WORKDIR /local-llm-server WORKDIR /local-llm-server
# The local local-llm-server repo may be cached, so we will fetch and reset to the remote every time. # The local local-llm-server repo may be cached, so we will fetch and reset to the remote every time.