happy0612 commited on
Commit
da7ea06
·
verified ·
1 Parent(s): 3ce111f

Add GeoNeXt Docker configuration

Browse files
Files changed (1) hide show
  1. Dockerfile +28 -0
Dockerfile ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
2
+
3
+ ENV DEBIAN_FRONTEND=noninteractive \
4
+ PYTHONUNBUFFERED=1 \
5
+ PIP_NO_CACHE_DIR=1 \
6
+ HF_HOME=/data/.huggingface
7
+
8
+ RUN apt-get update && apt-get install -y --no-install-recommends \
9
+ git python3 python3-pip python3-dev libgl1 libglib2.0-0 \
10
+ && rm -rf /var/lib/apt/lists/*
11
+
12
+ WORKDIR /app
13
+
14
+ RUN git clone --depth 1 https://github.com/Creative-Intelligence-Studio/GeoNeXt.git /app/GeoNeXt
15
+
16
+ RUN python3 -m pip install --upgrade pip \
17
+ && python3 -m pip install torch==2.3.1 torchvision==0.18.1 \
18
+ --index-url https://download.pytorch.org/whl/cu121 \
19
+ && python3 -m pip install -r /app/GeoNeXt/requirements-wan.txt \
20
+ && python3 -m pip install \
21
+ gradio==4.44.1 pydantic==2.10.6 fastapi==0.115.6 \
22
+ && python3 -m pip install -e /app/GeoNeXt
23
+
24
+ COPY app.py /app/app.py
25
+
26
+ EXPOSE 7860
27
+
28
+ CMD ["python3", "/app/app.py"]