1

docker pull cwq1913/lama-cleaner:gpu-0.33.0

1
2
3
4
5
6
7
8
9

mkdir lama && cd lama
mkdir -p {huggingface_cache,torch_cache}

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

1
2
3
4
5
6
7
8

lama$ cat docker_build_triton/Dockerfile.triton 
FROM cwq1913/lama-cleaner:gpu-0.33.0

RUN apt-get update && apt-get install -y --no-install-recommends \
        python3-pip && \
    rm -rf /var/lib/apt/lists/*

RUN pip3 install triton

1

docker build -t lama-cleaner-triton -f Dockerfile.triton .

1

docker run -d --restart unless-stopped --name lama-cleaner -p 8184:8184 -v $(pwd)/torch_cache:/root/.cache/torch -v $(pwd)/huggingface_cache:/root/.cache/huggingface --gpus all lama-cleaner-triton lama-cleaner --port=8184 --host=0.0.0.0 --gui