Sergey Mikhno commited on
Commit
1c1e490
·
1 Parent(s): 8356f0a

dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app/requirements.txt +1 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM ubuntu:22.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
 
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/cu113
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