-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDockerfile.txt
47 lines (38 loc) · 1.15 KB
/
Dockerfile.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# Dockerfile
FROM nvidia/cuda:11.2.2-devel-ubuntu20.04
ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=America/New_York
# Install Python and other required tools
RUN apt-get update && \
apt-get install -y --no-install-recommends \
git \
nano \
wget \
curl \
unzip\
cmake \
python3-dev \
cmake \
libopenblas-dev liblapack-dev \
libx11-dev \
&& rm -rf /var/lib/apt/lists/* &&\
ln -s /usr/bin/python3.8 /usr/bin/python
# Install pip
RUN wget https://bootstrap.pypa.io/get-pip.py && \
python get-pip.py && \
rm get-pip.py
# Install Python packages
COPY requirements.txt /tmp/
RUN pip install --extra-index-url https://download.pytorch.org/whl/cu116 \
torch==1.13.1 \
torchvision==0.14.1 \
-r /tmp/requirements.txt
# Install Ninja
RUN wget https://github.com/ninja-build/ninja/releases/download/v1.8.2/ninja-linux.zip && \
unzip ninja-linux.zip -d /usr/local/bin/ && \
update-alternatives --install /usr/bin/ninja ninja /usr/local/bin/ninja 1 --force && \
rm ninja-linux.zip
# Clone Git repository
RUN git clone https://github.com/minha12/IdDecoder.git /app
# Set the working directory
WORKDIR /app/