enskaff commited on
Commit
c392d1c
·
verified ·
1 Parent(s): 25bcaf8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +12 -2
Dockerfile CHANGED
@@ -1,7 +1,17 @@
1
  FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
2
 
3
- # System packages
4
- RUN apt-get update && apt-get install -y git python3 python3-pip && rm -rf /var/lib/apt/lists/*
 
 
 
 
 
 
 
 
 
 
5
 
6
  # Python packages
7
  RUN pip3 install vllm accelerate
 
1
  FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
2
 
3
+ # System packages (including curl and build tools)
4
+ RUN apt-get update && apt-get install -y \
5
+ git \
6
+ python3 \
7
+ python3-pip \
8
+ curl \
9
+ build-essential \
10
+ && rm -rf /var/lib/apt/lists/*
11
+
12
+ # Install Rust (needed for tokenizers)
13
+ RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
14
+ ENV PATH="/root/.cargo/bin:${PATH}"
15
 
16
  # Python packages
17
  RUN pip3 install vllm accelerate