add docker
This commit is contained in:
parent
ff4d1bb3b8
commit
11981aa2c4
|
@ -34,6 +34,11 @@
|
||||||
```
|
```
|
||||||
Tested with conda version 22.9.0
|
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
|
## 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)
|
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
64
docker/Dockerfile
Normal 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
14
docker/build_docker.sh
Normal 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
18
docker/run.sh
Normal 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
|
||||||
|
|
Loading…
Reference in a new issue