From 448ebcbe8d103269fbb56c8d29ca55886efcb07c Mon Sep 17 00:00:00 2001 From: Cyberes Date: Sat, 14 Oct 2023 20:41:40 -0600 Subject: [PATCH] docker: install pytorch 11.8 --- other/vllm/Docker/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/other/vllm/Docker/Dockerfile b/other/vllm/Docker/Dockerfile index 4064981..b4c0bda 100644 --- a/other/vllm/Docker/Dockerfile +++ b/other/vllm/Docker/Dockerfile @@ -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.