Update Dockerfile
Browse files- Dockerfile +3 -10
Dockerfile
CHANGED
@@ -1,14 +1,9 @@
|
|
1 |
-
|
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 |
-
|
10 |
RUN apt-get update
|
11 |
-
|
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"]
|