|
FROM nvidia/cuda:11.8.0-base-ubuntu22.04 |
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \ |
|
PYTHONUNBUFFERED=1 \ |
|
PYTHONIOENCODING=UTF-8 |
|
RUN --mount=type=cache,target=/var/cache/apt --mount=type=cache,target=/var/lib/apt apt update &&\ |
|
apt install -y \ |
|
wget \ |
|
git \ |
|
pkg-config \ |
|
python3 \ |
|
python3-pip \ |
|
python-is-python3 \ |
|
ffmpeg \ |
|
libnvrtc11.2 \ |
|
libtcmalloc-minimal4 |
|
|
|
RUN useradd -m -u 1000 ac |
|
RUN --mount=type=cache,target=/root/.cache python -m pip install --upgrade pip wheel |
|
ENV TORCH_COMMAND="pip install torch==2.0.1+cu118 torchaudio --extra-index-url https://download.pytorch.org/whl/cu118" |
|
RUN --mount=type=cache,target=/root/.cache python -m $TORCH_COMMAND |
|
RUN ln -s /usr/lib/x86_64-linux-gnu/libnvrtc.so.11.2 /usr/lib/x86_64-linux-gnu/libnvrtc.so |
|
USER 1000 |
|
RUN mkdir ~/.cache |
|
RUN --mount=type=cache,target=/home/ac/.cache --mount=source=.,target=/home/ac/audiocraft python -m pip install -r /home/ac/audiocraft/requirements.txt |
|
WORKDIR /home/ac/audiocraft |