PGSCOM commited on
Commit
c7981d2
verified
1 Parent(s): 44a4827

Dependencias vllm

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -13,6 +13,12 @@ WORKDIR /app
13
  COPY --chown=user . /app
14
  RUN pip install -e .
15
 
 
 
 
 
 
 
16
  # Clona el fork de vLLM de StarVector
17
  RUN git clone https://github.com/starvector/vllm.git /app/vllm
18
 
 
13
  COPY --chown=user . /app
14
  RUN pip install -e .
15
 
16
+ # Instala las dependencias de construcci贸n necesarias
17
+ RUN pip install numpy torch
18
+
19
+ # Establece la variable de entorno CUDA_HOME
20
+ ENV CUDA_HOME=/usr/local/cuda
21
+
22
  # Clona el fork de vLLM de StarVector
23
  RUN git clone https://github.com/starvector/vllm.git /app/vllm
24