docker: install pytorch 11.8
This commit is contained in:
parent
076501a784
commit
448ebcbe8d
|
@ -18,6 +18,8 @@ RUN rm /tmp/rathole.zip
|
|||
RUN cp /tmp/rathole /app
|
||||
|
||||
RUN python3 -m venv /venv
|
||||
RUN /venv/bin/pip install torch --index-url https://download.pytorch.org/whl/cu118
|
||||
|
||||
WORKDIR /local-llm-server
|
||||
|
||||
# The local local-llm-server repo may be cached, so we will fetch and reset to the remote every time.
|
||||
|
|
Reference in New Issue