From 11981aa2c4da9d28a72c4bb31ead0f004113aefa Mon Sep 17 00:00:00 2001 From: xzeng Date: Mon, 20 Feb 2023 00:27:19 -0500 Subject: [PATCH] add docker --- README.md | 5 ++++ docker/Dockerfile | 64 ++++++++++++++++++++++++++++++++++++++++++ docker/build_docker.sh | 14 +++++++++ docker/run.sh | 18 ++++++++++++ 4 files changed, 101 insertions(+) create mode 100644 docker/Dockerfile create mode 100644 docker/build_docker.sh create mode 100644 docker/run.sh diff --git a/README.md b/README.md index 94ed9e6..ce63cfe 100644 --- a/README.md +++ b/README.md @@ -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) diff --git a/docker/Dockerfile b/docker/Dockerfile new file mode 100644 index 0000000..4c5f4b6 --- /dev/null +++ b/docker/Dockerfile @@ -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 diff --git a/docker/build_docker.sh b/docker/build_docker.sh new file mode 100644 index 0000000..58cfe3e --- /dev/null +++ b/docker/build_docker.sh @@ -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 . + diff --git a/docker/run.sh b/docker/run.sh new file mode 100644 index 0000000..9875cd6 --- /dev/null +++ b/docker/run.sh @@ -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 +