Spaces:
Running
on
Zero
Running
on
Zero
add dockerfile
Browse files- dockerfile +13 -0
- packages.txt +0 -12
- requirements.txt +0 -49
dockerfile
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM huggingface/transformers-pytorch-gpu
|
| 2 |
+
|
| 3 |
+
RUN apt-get update
|
| 4 |
+
RUN apt-get -y install libglfw3-dev libgles2-mesa-dev libgl1 freeglut3-dev unzip ffmpeg libsm6 libxext6 libgl1-mesa-dri libegl1-mesa libgbm1 build-essential
|
| 5 |
+
RUN python3 -m pip install --upgrade pip
|
| 6 |
+
RUN pip3 install opencv-python numpy trimesh plyfile pyyaml scikit-image scikit-learn chumpy tensorboard kornia loguru pycocotools yacs lmdb fire setuptools einops tqdm ipython gym transformers==4.44.2 OmegaConf matplotlib diffusers rembg segment_anything jaxtyping imageio iopath timm open3d pyrender pytorch-lightning smplx==0.1.28 chumpy @ git+https://github.com/zerchen/chumpy xtcocotools pandas hydra-core hydra-submitit-launcher hydra-colorlog pyrootutils rich webdataset ultralytics dill git+https://github.com/facebookresearch/pytorch3d.git@stable git+https://github.com/zerchen/lang-segment-anything.git typeguard gradio==5.0.2
|
| 7 |
+
|
| 8 |
+
RUN wget -q https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run -o /tmp/cuda121.run
|
| 9 |
+
RUN chmod +x /tmp/cuda121.run && /tmp/cuda121.run --silent --toolkit
|
| 10 |
+
RUN cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app
|
| 11 |
+
|
| 12 |
+
ENTRYPOINT ["python3"]
|
| 13 |
+
CMD ["app.py"]
|
packages.txt
DELETED
|
@@ -1,12 +0,0 @@
|
|
| 1 |
-
libglfw3-dev
|
| 2 |
-
libgles2-mesa-dev
|
| 3 |
-
libgl1
|
| 4 |
-
freeglut3-dev
|
| 5 |
-
unzip
|
| 6 |
-
ffmpeg
|
| 7 |
-
libsm6
|
| 8 |
-
libxext6
|
| 9 |
-
libgl1-mesa-dri
|
| 10 |
-
libegl1-mesa
|
| 11 |
-
libgbm1
|
| 12 |
-
build-essential
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
requirements.txt
DELETED
|
@@ -1,49 +0,0 @@
|
|
| 1 |
-
opencv-python
|
| 2 |
-
numpy
|
| 3 |
-
trimesh
|
| 4 |
-
plyfile
|
| 5 |
-
pyyaml
|
| 6 |
-
scikit-image
|
| 7 |
-
scikit-learn
|
| 8 |
-
chumpy
|
| 9 |
-
tensorboard
|
| 10 |
-
kornia
|
| 11 |
-
loguru
|
| 12 |
-
pycocotools
|
| 13 |
-
yacs
|
| 14 |
-
lmdb
|
| 15 |
-
fire
|
| 16 |
-
setuptools
|
| 17 |
-
einops
|
| 18 |
-
tqdm
|
| 19 |
-
ipython
|
| 20 |
-
gym
|
| 21 |
-
transformers==4.44.2
|
| 22 |
-
OmegaConf
|
| 23 |
-
matplotlib
|
| 24 |
-
diffusers
|
| 25 |
-
rembg
|
| 26 |
-
segment_anything
|
| 27 |
-
jaxtyping
|
| 28 |
-
imageio
|
| 29 |
-
iopath
|
| 30 |
-
timm
|
| 31 |
-
open3d
|
| 32 |
-
pyrender
|
| 33 |
-
pytorch-lightning
|
| 34 |
-
smplx==0.1.28
|
| 35 |
-
chumpy @ git+https://github.com/zerchen/chumpy
|
| 36 |
-
xtcocotools
|
| 37 |
-
pandas
|
| 38 |
-
hydra-core
|
| 39 |
-
hydra-submitit-launcher
|
| 40 |
-
hydra-colorlog
|
| 41 |
-
pyrootutils
|
| 42 |
-
rich
|
| 43 |
-
webdataset
|
| 44 |
-
ultralytics
|
| 45 |
-
dill
|
| 46 |
-
git+https://github.com/facebookresearch/pytorch3d.git@stable
|
| 47 |
-
git+https://github.com/zerchen/lang-segment-anything.git
|
| 48 |
-
typeguard
|
| 49 |
-
gradio==5.0.2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|