ahmedzein commited on
Commit
bf08476
·
verified ·
1 Parent(s): a3691b9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -10
Dockerfile CHANGED
@@ -1,14 +1,9 @@
1
- # Use the official Python 3.10.9 image
2
- FROM ubuntu
3
- # RUN adduser
4
- RUN useradd -m -s /bin/bash xyz
5
- RUN mkdir -p /usr/src/app/
6
- RUN chown -R xyz:xyz /usr/src/app/
7
 
8
  WORKDIR /usr/src/app
9
- # RUN add-apt-repository main
10
  RUN apt-get update
11
- # RUN apt-get install -y libgl1-mesa-glx
12
  RUN apt install -y python3-pip
13
  RUN rm /usr/lib/python3.*/EXTERNALLY-MANAGED
14
 
@@ -23,6 +18,4 @@ ENV HF_HOME=/tmp
23
  ENV TRANSFORMERS_CACHE=/tmp
24
 
25
 
26
-
27
- # Start the FastAPI app on port 7860, the default port expected by Spaces
28
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
1
+ FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
 
 
 
 
 
2
 
3
  WORKDIR /usr/src/app
4
+
5
  RUN apt-get update
6
+
7
  RUN apt install -y python3-pip
8
  RUN rm /usr/lib/python3.*/EXTERNALLY-MANAGED
9
 
 
18
  ENV TRANSFORMERS_CACHE=/tmp
19
 
20
 
 
 
21
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]