Spaces:
Running
on
T4
Running
on
T4
Update Dockerfile
Browse files- Dockerfile +5 -5
Dockerfile
CHANGED
|
@@ -1,4 +1,8 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 4 |
ENV PYTHONUNBUFFERED=1
|
|
@@ -22,10 +26,6 @@ COPY ./requirements.txt /code/requirements.txt
|
|
| 22 |
|
| 23 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 24 |
|
| 25 |
-
RUN mim install mmengine
|
| 26 |
-
RUN mim install mmcv
|
| 27 |
-
RUN mim install mmdet
|
| 28 |
-
RUN mim install mmocr
|
| 29 |
|
| 30 |
# Set up a new user named "user" with user ID 1000
|
| 31 |
RUN useradd -m -u 1000 user
|
|
|
|
| 1 |
+
ARG PYTORCH="1.9.0"
|
| 2 |
+
ARG CUDA="11.1"
|
| 3 |
+
ARG CUDNN="8"
|
| 4 |
+
|
| 5 |
+
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
|
| 6 |
|
| 7 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 8 |
ENV PYTHONUNBUFFERED=1
|
|
|
|
| 26 |
|
| 27 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# Set up a new user named "user" with user ID 1000
|
| 31 |
RUN useradd -m -u 1000 user
|