Kangarroar commited on
Commit
b099b78
1 Parent(s): 5272027

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +31 -5
Dockerfile CHANGED
@@ -1,17 +1,43 @@
1
- FROM basisai/python-cuda:3.8.6-10.1
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  ARG DEBIAN_FRONTEND=noninteractive
4
  ENV PYTHONUNBUFFERED=1
 
 
5
 
6
  WORKDIR /app
7
-
 
 
 
 
 
 
 
8
  COPY ./requirements.txt /app/requirements.txt
9
  COPY ./packages.txt /app/packages.txt
10
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
11
- RUN pip install --no-cache-dir -r /app/requirements.txt
12
- RUN pip install --no-cache-dir numba==0.56.3
 
 
 
 
 
 
13
  RUN pip install --upgrade pip setuptools wheel
14
- RUN pip install pyworld
15
  RUN pip install soundfile
16
 
17
  WORKDIR /app
 
1
+ FROM nvidia/cuda:11.8-devel-ubuntu18.04
2
+
3
+ RUN apt-get update && apt-get install -y --no-install-recommends \
4
+ cuda-libraries-11-8=11.8.2-1 \
5
+ cuda-nvtx-11-8=11.8.2-1 \
6
+ libcudnn8=8.0.4.30-1+cuda11.0 \
7
+ && \
8
+ apt-mark hold cuda-libraries-11-8 cuda-nvtx-11-8 libcudnn8 && \
9
+ rm -rf /var/lib/apt/lists/*
10
+
11
+ ENV PATH /usr/local/cuda-11.8/bin:$PATH
12
+ ENV LD_LIBRARY_PATH /usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
13
 
14
  ARG DEBIAN_FRONTEND=noninteractive
15
  ENV PYTHONUNBUFFERED=1
16
+ ENV PYTHON_INCLUDE /usr/include/python3.8
17
+ ENV PYTHON_LIB /usr/lib/x86_64-linux-gnu/libpython3.8.so
18
 
19
  WORKDIR /app
20
+ # Install python, git & ffmpeg
21
+ RUN apt-get update && apt-get install --no-install-recommends -y \
22
+ build-essential \
23
+ python3.9 \
24
+ python3-pip \
25
+ git \
26
+ ffmpeg \
27
+ && apt-get clean && rm -rf /var/lib/apt/lists/*
28
  COPY ./requirements.txt /app/requirements.txt
29
  COPY ./packages.txt /app/packages.txt
30
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
31
+ RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
32
+ RUN apt-get update && apt-get install -y python3-dev
33
+ RUN apt-get update && apt-get install -y build-essential
34
+ RUN which python3
35
+ RUN which python3-config
36
+ RUN pip3 install --no-cache-dir -r /app/requirements.txt
37
+ RUN apt-get update && apt-get install -y build-essential
38
+ RUN pip3 install --no-cache-dir numba==0.56.3
39
  RUN pip install --upgrade pip setuptools wheel
40
+ RUN pip3 install --no-binary :all: pyworld
41
  RUN pip install soundfile
42
 
43
  WORKDIR /app