Sergey Mikhno
commited on
Commit
·
1c1e490
1
Parent(s):
8356f0a
dockerfile
Browse files- Dockerfile +1 -1
- app/requirements.txt +1 -1
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM
|
2 |
ENV DEBIAN_FRONTEND=noninteractive
|
3 |
RUN apt-get update && apt-get install -y build-essential && apt-get install -y libopencv-dev && apt-get install -y python3.10 python3-pip
|
4 |
#FROM python:3.9
|
|
|
1 |
+
FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
|
2 |
ENV DEBIAN_FRONTEND=noninteractive
|
3 |
RUN apt-get update && apt-get install -y build-essential && apt-get install -y libopencv-dev && apt-get install -y python3.10 python3-pip
|
4 |
#FROM python:3.9
|
app/requirements.txt
CHANGED
@@ -2,7 +2,7 @@ doclayout-yolo == 0.0.4
|
|
2 |
flask == 3.1.1
|
3 |
huggingface-hub == 0.33.0
|
4 |
gunicorn
|
5 |
-
--extra-index-url https://download.pytorch.org/whl/
|
6 |
torch
|
7 |
spaces
|
8 |
|
|
|
2 |
flask == 3.1.1
|
3 |
huggingface-hub == 0.33.0
|
4 |
gunicorn
|
5 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
6 |
torch
|
7 |
spaces
|
8 |
|