fomext commited on
Commit
18932ad
·
verified ·
1 Parent(s): 606c049

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +15 -5
Dockerfile CHANGED
@@ -1,9 +1,15 @@
1
- FROM python:3.10
2
 
3
- # System deps (FFmpeg dev headers REQUIRED for PyAV)
 
 
 
4
  RUN apt-get update && apt-get install -y \
5
- ffmpeg \
 
 
6
  git \
 
7
  pkg-config \
8
  libgl1 \
9
  libglib2.0-0 \
@@ -16,6 +22,10 @@ RUN apt-get update && apt-get install -y \
16
  libswresample-dev \
17
  && rm -rf /var/lib/apt/lists/*
18
 
 
 
 
 
19
  # User
20
  RUN useradd -m -u 1000 user
21
  USER user
@@ -25,13 +35,13 @@ WORKDIR /app
25
  # Upgrade pip
26
  RUN pip install --upgrade pip setuptools wheel
27
 
28
- # Install torch FIRST (CPU)
29
  RUN pip install \
30
  torch==2.1.0+cpu \
31
  torchaudio==2.1.0+cpu \
32
  --index-url https://download.pytorch.org/whl/cpu
33
 
34
- # Install audiocraft (will now succeed)
35
  RUN pip install \
36
  git+https://github.com/facebookresearch/audiocraft.git@v1.3.0
37
 
 
1
+ FROM ubuntu:22.04
2
 
3
+ ENV DEBIAN_FRONTEND=noninteractive
4
+ ENV PYTHONUNBUFFERED=1
5
+
6
+ # System deps (FFmpeg 4.4 compatible)
7
  RUN apt-get update && apt-get install -y \
8
+ python3.10 \
9
+ python3.10-dev \
10
+ python3-pip \
11
  git \
12
+ ffmpeg \
13
  pkg-config \
14
  libgl1 \
15
  libglib2.0-0 \
 
22
  libswresample-dev \
23
  && rm -rf /var/lib/apt/lists/*
24
 
25
+ # Make python default
26
+ RUN ln -s /usr/bin/python3.10 /usr/bin/python
27
+ RUN ln -s /usr/bin/pip3 /usr/bin/pip
28
+
29
  # User
30
  RUN useradd -m -u 1000 user
31
  USER user
 
35
  # Upgrade pip
36
  RUN pip install --upgrade pip setuptools wheel
37
 
38
+ # Install torch (CPU)
39
  RUN pip install \
40
  torch==2.1.0+cpu \
41
  torchaudio==2.1.0+cpu \
42
  --index-url https://download.pytorch.org/whl/cpu
43
 
44
+ # Install audiocraft (NOW SUCCEEDS)
45
  RUN pip install \
46
  git+https://github.com/facebookresearch/audiocraft.git@v1.3.0
47