From f9b44f69ad2c6415629794991333845d08ca8839 Mon Sep 17 00:00:00 2001 From: Bryan Thornbury Date: Wed, 16 Aug 2023 16:39:37 -0700 Subject: [PATCH] Dockerfile GPU Support Confirmed working on 3090ti --- Dockerfile | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 00b1196a..2cdaf4df 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,10 @@ -FROM debian:stable +# FROM nvidia/cuda:11.6.2-base-ubuntu20.04 +FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 -# Install system packages -RUN apt update && apt install -y git pip +ENV LD_LIBRARY_PATH=/usr/local/cuda/efa/lib:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda:/usr/lib/x86_64-linux-gnu + +RUN apt update && apt install -qqy python3 python3-virtualenv python3-dev \ + build-essential python3-pip git vim nano wget curl git-lfs ffmpeg # Create non-root user RUN useradd -m -d /bark bark @@ -27,6 +30,10 @@ RUN pip install -r requirements.txt RUN sed -i "s/server_name: ''/server_name: 0.0.0.0/g" ./config.yaml # Suggested volumes +RUN mkdir -p /bark/bark-gui/assets/prompts/custom +RUN mkdir -p /bark/bark-gui/models +RUN mkdir -p /bark/.cache/huggingface/hub + VOLUME /bark/bark-gui/assets/prompts/custom VOLUME /bark/bark-gui/models VOLUME /bark/.cache/huggingface/hub