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 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.
|
||||||
|
|
Reference in New Issue