Spaces:
Paused
Paused
Commit
·
d3fc1c3
1
Parent(s):
f18e25c
Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -1,7 +1,10 @@
|
|
| 1 |
-
FROM
|
| 2 |
|
| 3 |
WORKDIR /code
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
COPY ./requirements.txt /code/requirements.txt
|
| 6 |
|
| 7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04
|
| 2 |
|
| 3 |
WORKDIR /code
|
| 4 |
|
| 5 |
+
RUN apt-get update && \
|
| 6 |
+
apt-get install -y python3 python3-pip
|
| 7 |
+
|
| 8 |
COPY ./requirements.txt /code/requirements.txt
|
| 9 |
|
| 10 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|