makakwastaken commited on
Commit
b377d7a
·
1 Parent(s): 1e36d64

Allow GPU usage (Future proof)

Browse files
Files changed (1) hide show
  1. Dockerfile +47 -0
Dockerfile ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04
2
+
3
+ RUN apt-get update
4
+ RUN apt-get install software-properties-common -y
5
+ RUN add-apt-repository ppa:deadsnakes/ppa -y
6
+ RUN apt-get install python3.8 -y
7
+ RUN apt-get install python3.8-distutils -y
8
+
9
+ RUN apt-get install python-dev python3-dev python3.8-dev \
10
+ build-essential libssl-dev libffi-dev \
11
+ libxml2-dev libxslt1-dev zlib1g-dev -y
12
+
13
+ RUN apt-get update && \
14
+ apt-get install -y --no-install-recommends gcc
15
+
16
+ RUN apt-get install curl wget -y
17
+
18
+ RUN apt-get install linux-libc-dev -y
19
+
20
+ RUN apt-get install libc6-dev -y
21
+
22
+ RUN apt-get install git -y
23
+
24
+ RUN wget https://bootstrap.pypa.io/get-pip.py
25
+ RUN python3.8 get-pip.py
26
+
27
+ # Set the working directory to /app
28
+ WORKDIR /
29
+
30
+ # Get CutLER
31
+ RUN git clone --recursive https://github.com/Ad-Visual/CutLER
32
+
33
+ # Install CutLER
34
+ RUN pip install --no-cache-dir --upgrade -r /CutLER/requirements.txt
35
+
36
+
37
+ # Copy the requirements to the container at /app
38
+ COPY ./app/requirements.txt /requirements.txt
39
+
40
+ # Install any needed packages specified in requirements.txt
41
+ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
42
+
43
+ #
44
+ COPY ./app /
45
+
46
+ #
47
+ CMD ["python", "main.py"]