banao-tech commited on
Commit
5cd2744
·
verified ·
1 Parent(s): 2aa9adc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -8,8 +8,11 @@ RUN chmod 1777 /tmp \
8
  && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
9
  && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
10
  && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
 
 
 
11
 
12
- RUN pip install fastapi[all]
13
 
14
 
15
  COPY main.py main.py
 
8
  && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
9
  && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
10
  && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
11
+
12
+ # Copy and install Python requirements
13
+ COPY requirements.txt .
14
+ RUN pip install --no-cache-dir -r requirements.txt
15
 
 
16
 
17
 
18
  COPY main.py main.py