Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
FROM python:3.11-slim-bullseye
|
| 2 |
-
|
| 3 |
RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-recommends \
|
| 4 |
python3 \
|
| 5 |
python3-pip \
|
|
|
|
| 1 |
+
#FROM python:3.11-slim-bullseye
|
| 2 |
+
FROM huggingface/optimum-nvidia
|
| 3 |
RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-recommends \
|
| 4 |
python3 \
|
| 5 |
python3-pip \
|