Safran/LION
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add docker

Browse source
This commit is contained in:
xzeng 2023-02-20 00:27:19 -05:00
parent ff4d1bb3b8
commit 11981aa2c4
4 changed files with 101 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
README.md
View file

@ -34,6 +34,11 @@
```
Tested with conda version 22.9.0
* Using Docker
* build the docker with `bash ./docker/build_docker.sh`
* launch the docker with `bash ./docker/run.sh`
## Demo
run `python demo.py`, will load the released text2shape model on hugging face and generate a chair point cloud. (Note: the checkpoint is not released yet, the files loaded in the `demo.py` file is not available at this point)

64
docker/Dockerfile Normal file
View file

@ -0,0 +1,64 @@
# ---------------------------------------------------------------
# Copyright (c) 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
# ---------------------------------------------------------------
FROM nvcr.io/nvidia/pytorch:21.12-py3
ARG PYTHON_VERSION=3.8
ARG WITH_TORCHVISION=1
RUN python -m pip uninstall torch -y
RUN python -m pip uninstall torchvision -y
RUN python -m pip uninstall torchaudio -y
RUN python -m pip uninstall torchtext -y
RUN python -m pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
RUN apt-get update
RUN apt-get install -y build-essential zsh tmux wget
RUN apt-get install -y libgl1-mesa-glx libusb-1.0-0-dev
RUN apt purge --auto-remove cmake
RUN python -m pip install cmake
RUN python -m pip install scipy
#RUN python -m pip install pykeops
RUN python -m pip install h5py
RUN python -m pip install wandb
RUN python -m pip install --ignore-installed PyYAML
#RUN python -m pip install open3d
RUN python -m pip install cupy-cuda111
RUN python -m pip install nestargs
RUN python -m pip install flatten_dict
#RUN python -m pip uninstall point_cloud_utils
#RUN python -m pip install git+https://github.com/fwilliams/point-cloud-utils.git
RUN conda install -y scikit-image
RUN apt-get update
#RUN python -m pip uninstall torch-scatter
RUN export PATH=/usr/local/cuda-11.1/bin:$PATH
RUN export CPATH=/usr/local/cuda-11.1/include:$CPATH
RUN export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
RUN export DYLD_LIBRARY_PATH=/usr/local/cuda-11.1/lib:$DYLD_LIBRARY_PATH
RUN python -m pip install git+https://github.com/openai/CLIP.git
RUN python -m pip install loguru einops comet_ml calmsize diffusers
#COPY environment.yml .
#COPY activate.sh .
#
#RUN apt -y update && apt -y install wget curl vim libgcrypt20 coreutils libgl1-mesa-glx
#
#RUN wget https://repo.continuum.io/miniconda/Miniconda3-3.7.0-Linux-x86_64.sh -O ./miniconda.sh
#
#RUN chmod ouga+xw ./miniconda.sh
#RUN bash ./miniconda.sh -b -p ./miniconda
#
#ENV PATH /opt/pytorch/miniconda/bin:$PATH
#
#RUN conda update conda
#RUN conda env create -f environment.yml
## RUN /bin/bash -c "source activate pvcnn_pyt13"
#RUN conda activate pvcnn_pyt13
#RUN pip install einops
#RUN pip install sklearn
#RUN pip install tensorflow

14
docker/build_docker.sh Normal file
View file

@ -0,0 +1,14 @@
# ---------------------------------------------------------------
# Copyright (c) 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
# ---------------------------------------------------------------
# if permission error, run
# newgrp docker
docker build -t nvcr.io/nvidian/lion_env:0 .

18
docker/run.sh Normal file
View file

@ -0,0 +1,18 @@
#!/bin/bash
# ---------------------------------------------------------------
# Copyright (c) 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
# ---------------------------------------------------------------
docker="nvcr.io/nvidian/lion_env:0"
echo $PWD
exit
code_path=$PWD
docker run --gpus all -p 8081:8081 --ipc=host -v $code_path:$code_path -it $docker bash