changed to docker

Dockerfile

@@ -0,0 +1,49 @@
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+
+ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV TZ=America/Los_Angeles
+RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    git-lfs \
+    python3 \
+    python3-pip \
+    python3-setuptools \
+    python3-wheel \
+    python3-dev \
+    cmake \
+    unzip \
+    vim \
+    ffmpeg \
+    curl \
+    wget \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    build-essential \
+    libsparsehash-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set Python3 as the default version
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1
+
+RUN pip3 install pip=
+
+# Install PyTorch
+RUN pip3 install --no-cache-dir \
+    torch \
+    torchvision \
+    torchaudio \
+    --index-url https://download.pytorch.org/whl/cu118
+
+COPY requirements.txt /opt/app/requirements.txt
+WORKDIR /opt/app
+RUN pip install -r requirements.txt
+# COPY . /opt/app
+COPY requirements.txt /opt/app
+
+ARG TORCH_CUDA_ARCH_LIST="5.2;7.0;7.2;8.0;8.6+PTX"
+ARG IABN_FORCE_CUDA="1"
+ARG FORCE_CUDA="1"
+RUN pip3 install --no-cache-dir \
+    inplace_abn \
+    git+https://github.com/mit-han-lab/[email protected]

README.md

@@ -3,9 +3,8 @@ title: One-2-3-45
 emoji: 📸🚀🌟
 colorFrom: red
 colorTo: yellow
-sdk: gradio
-sdk_version: 4.22.0
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: true
 license: mit
 ---

app.py

@@ -11,15 +11,15 @@ sys.path.append(code_dir)
 elev_est_dir = os.path.abspath(os.path.join(code_dir, "elevation_estimate"))
 sys.path.append(elev_est_dir)
 
-if not is_local_run:
+# if not is_local_run:
     # export TORCH_CUDA_ARCH_LIST="7.0;7.2;8.0;8.6"
     # export IABN_FORCE_CUDA=1
-    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    os.environ["IABN_FORCE_CUDA"] = "1"
-    os.environ["FORCE_CUDA"] = "1"
-    subprocess.run(['pip', 'install', 'inplace_abn'])
+    # os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
+    # os.environ["IABN_FORCE_CUDA"] = "1"
+    # os.environ["FORCE_CUDA"] = "1"
+    # subprocess.run(['pip', 'install', 'inplace_abn'])
     # FORCE_CUDA=1 pip install --no-cache-dir git+https://github.com/mit-han-lab/[email protected]
-    subprocess.run(['pip', 'install', '--no-cache-dir', 'git+https://github.com/mit-han-lab/[email protected]'])
+    # subprocess.run(['pip', 'install', '--no-cache-dir', 'git+https://github.com/mit-han-lab/[email protected]'])
 
 import shutil
 import torch

requirements.txt

@@ -1,4 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
+gradio==4.38.1
 torch==2.0.1
 torchvision
 albumentations>=0.4.3
@@ -17,7 +18,6 @@ kornia>=0.6
 webdataset>=0.2.5
 torchmetrics>=0.6.0
 fire>=0.4.0
-gradio>=3.41.1
 diffusers>=0.12.1
 datasets[vision]>=2.4.0
 rich>=13.3.2