optigesr commited on
Commit
5cf3651
·
1 Parent(s): 1905bb9

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +32 -0
Dockerfile ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:11.3.1-base-ubuntu20.04
2
+
3
+ ENV PYTHON_VERSION=3.8
4
+
5
+ RUN export DEBIAN_FRONTEND=noninteractive \
6
+ && apt-get -qq update \
7
+ && apt-get -qq install --no-install-recommends \
8
+ libsndfile1-dev \
9
+ git \
10
+ python${PYTHON_VERSION} \
11
+ python${PYTHON_VERSION}-venv \
12
+ python3-pip \
13
+ && rm -rf /var/lib/apt/lists/*
14
+
15
+ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
16
+ ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \
17
+ ln -s -f /usr/bin/pip3 /usr/bin/pip
18
+
19
+ RUN pip install --upgrade pip
20
+
21
+ # 2. Copy files
22
+ COPY . /src
23
+
24
+ RUN pip install torch==1.12.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
25
+
26
+ WORKDIR /src
27
+ # 3. Install dependencies
28
+ RUN pip install -r requirements-docker.txt
29
+
30
+ RUN python3 setup.py install
31
+
32
+