Merge pull request #15 from tensorpix/feature/optimizeImage

Set ubuntu22.04 as base image; Bump torch and lighting

.github/workflows/linter.yml

@@ -10,18 +10,17 @@ jobs:
     name: Run pre-commit hooks
     runs-on: ubuntu-latest
 
-    permissions:
-      contents: read
-      packages: read
-      statuses: write
-
     steps:
-      - name: Checkout Code
+      - name: Checkout code
         uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
 
-      - uses: pre-commit-ci/lite-action@v1.0.1
-        if: always()
+      - name: Set up Python
+        uses: actions/setup-python@v3
         with:
-          msg: apply pre-commit hooks
+          python-version: "3.10"
+
+      - name: Install pre-commit
+        run: pip install -r requirements-dev.txt
+
+      - name: Run pre-commit checks
+        run: pre-commit run --all-files

dockerfiles/cuda118/Dockerfile

@@ -1,14 +1,12 @@
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM ubuntu:22.04
 
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
 	python3 \
+	python3-pip \
 	curl && \
 	apt clean && \
 	rm -rf /var/lib/apt/lists/*
 
-RUN curl https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py && \
-	python3 /tmp/get-pip.py
-
 COPY requirements.txt /tmp/requirements.txt
 RUN pip3 install --no-cache-dir -r /tmp/requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118
 

dockerfiles/cuda120/Dockerfile

@@ -1,14 +1,12 @@
-FROM nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04
+FROM ubuntu:22.04
 
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
 	python3 \
+	python3-pip \
 	curl && \
 	apt clean && \
 	rm -rf /var/lib/apt/lists/*
 
-RUN curl https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py && \
-	python3 /tmp/get-pip.py
-
 COPY requirements.txt /tmp/requirements.txt
 RUN pip3 install --no-cache-dir -r /tmp/requirements.txt
 

requirements-dev.txt

@@ -1 +1 @@
-pre-commit~=3.5.0
+pre-commit==3.*

requirements.txt

@@ -1,6 +1,6 @@
-lightning==2.1.2
+lightning==2.1.4
 protobuf==3.20.*
 segmentation-models-pytorch==0.3.3
 six==1.16.0
-torch==2.1.1
-torchvision==0.16.1
+torch==2.1.2
+torchvision==0.16.2