DOcker_content

Author: AvTheBlackBird

Dockerfile

@@ -0,0 +1,35 @@
+# syntax=docker/dockerfile:1
+
+# Stage 1: Base image
+FROM postgres
+
+ENV DEBIAN_FRONTEND=noninteractive POSTGRES_PASSWORD=test1234
+
+WORKDIR /
+
+# Install dependencies
+RUN apt-get update \
+    && apt-get install -y git tmux nala libpq-dev python3.11-dev nginx rabbitmq-server libpq-dev python3.11-venv
+
+RUN git clone https://github.com/smlab-niser/tirtha-public.git && cd /tirtha-public && git submodule update --init --recursive 
+RUN python3.11 -m venv venv \
+    && nala install -y python3-pip && /venv/bin/pip install --upgrade pip setuptools wheel \
+    && cd /tirtha-public && /venv/bin/pip install -r ./requirements.txt --default-timeout=2000 \ 
+    && /venv/bin/pip install -e /tirtha-public/tirtha_bk/nn_models/nsfw_model/ && /venv/bin/pip install protobuf==3.20.3 && nala install -y wget unzip && wget https://smlab.niser.ac.in/project/tirtha/static/artifacts/MR2021.1.0.zip \
+    && unzip MR2021.1.0.zip && mv ./bin21/ /tirtha-public/tirtha_bk/bin21/ && rm ./MR2021.1.0.zip && wget https://smlab.niser.ac.in/project/tirtha/static/artifacts/ckpt_kadid10k.pt \
+    && mv ./ckpt_kadid10k.pt /tirtha-public/tirtha_bk/nn_models/MANIQA/
+
+RUN nala install -y npm systemctl nano && npm install -g obj2gltf && wget https://github.com/zeux/meshoptimizer/releases/download/v0.20/gltfpack-ubuntu.zip \
+    && unzip gltfpack-ubuntu.zip && chmod +x gltfpack && mv gltfpack /usr/local/bin/ && rm /gltfpack-ubuntu.zip
+
+RUN mv /tirtha-public/tirtha_bk/config/tirtha.docker.nginx /tirtha-public/tirtha_bk/config/tirtha.nginx \
+    && mv /tirtha-public/tirtha_bk/tirtha_bk/local_settings.docker.py /tirtha-public/tirtha_bk/tirtha_bk/local_settings.py \
+    && mv /tirtha-public/tirtha_bk/gunicorn/gunicorn.conf.docker.py /tirtha-public/tirtha_bk/gunicorn/gunicorn.conf.py
+
+# COPY ./tirtha-public/tirtha_bk/gunicorn/gunicorn.conf.py /tirtha-public/tirtha_bk/gunicorn/    
+
+COPY ./init_db.sh /docker-entrypoint-initdb.d
+# COPY ./tirtha /var/www/tirtha 
+COPY ./start.sh /
+
+RUN chmod +x /docker-entrypoint-initdb.d/init_db.sh && chmod +x /start.sh

compose.yml

@@ -0,0 +1,18 @@
+services:
+  cuda:
+    container_name: cuda
+    image: nvidia/cuda:12.3.1-runtime-ubuntu22.04
+    restart: always
+  tirtha_repo:
+    container_name: tirtha_repo
+    build: .
+    restart: always
+    volumes:
+      - type: bind
+        source: ./tirtha
+        target: /var/www/tirtha 
+    ports:
+      - "8000:8000"
+      - "15672:15672"
+    depends_on:
+      - cuda

init_db.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+set -e
+
+psql -v ON_ERROR_STOP=1 --username "postgres" <<-EOSQL
+	CREATE DATABASE dbtirtha;
+        CREATE USER dbtirthauser WITH PASSWORD 'docker';
+        ALTER ROLE dbtirthauser SET client_encoding TO 'utf8';
+        ALTER ROLE dbtirthauser SET default_transaction_isolation TO 'read committed';
+        ALTER ROLE dbtirthauser SET timezone TO 'UTC';
+        ALTER DATABASE dbtirtha OWNER TO dbtirthauser;
+        GRANT ALL PRIVILEGES ON DATABASE dbtirtha TO dbtirthauser;
+        GRANT CREATE ON SCHEMA public TO dbtirthauser;
+EOSQL

start.sh

@@ -0,0 +1,23 @@
+# 
+cp /tirtha-public/tirtha_bk/config/tirtha.nginx /etc/nginx/sites-available/tirtha 
+ln -s /etc/nginx/sites-available/tirtha /etc/nginx/sites-enabled/
+systemctl enable rabbitmq-server
+systemctl start rabbitmq-server
+rabbitmq-plugins enable rabbitmq_management
+rabbitmqctl add_user rmqtirthauser rmqtirthapwd
+rabbitmqctl add_vhost rmqtirtha
+rabbitmqctl set_user_tags rmqtirthauser administrator
+rabbitmqctl set_permissions -p rmqtirtha rmqtirthauser ".*" ".*" ".*"
+rabbitmqctl eval 'application:get_env(rabbit, consumer_timeout).'
+echo "consumer_timeout = 31556952000" | tee -a /etc/rabbitmq/rabbitmq.conf
+systemctl restart rabbitmq-server
+cp tirtha-public/tirtha_bk/tirtha/templates/tirtha/403.html /var/www/tirtha/errors/
+cp tirtha-public/tirtha_bk/tirtha/templates/tirtha/404.html /var/www/tirtha/errors/
+cp tirtha-public/tirtha_bk/tirtha/templates/tirtha/500.html /var/www/tirtha/errors/
+cp tirtha-public/tirtha_bk/tirtha/templates/tirtha/503.html /var/www/tirtha/errors/
+chmod -R 755 /var/www/tirtha/
+chown -R $(whoami):$(whoami) /var/www/tirtha/
+source /venv/bin/activate
+python /tirtha-public/tirtha_bk/manage.py makemigrations tirtha
+python /tirtha-public/tirtha_bk/manage.py collectstatic
+python /tirtha-public/tirtha_bk/manage.py migrate