[Docker] [Experiments] NVIDIA Kaolin Wisp配置

Installing Kaolin Wisp

Reference: Kaolin Wisp installation

1. Docker (Python 3.8 environment)

docker run --gpus all --ipc=host --net=host --pid=host --name vqad -v /public/data0/MULT:/MULT -v /etc/localtime:/etc/localtime -v /public/data0/data0:/data0 -v /public/data1:/data1 -v /raid:/raid -it nvcr.io/nvidia/pytorch:21.08-py3-pt1.10dvgo /bin/bash

docker start vqad

docker exec -it vqad bash

pip install --upgrade pip

2. Install OpenEXR (Linux) (root)

apt-get update

apt-get install libopenexr-dev

3. Install Kaolin

Reference: Kaolin installation

git clone --recursive https://github.com/NVIDIAGameWorks/kaolin

cd kaolin

python setup.py develop

Testing

Run a quick test of your installation and version:

python -c “import kaolin; print(kaolin.version)”

For an exhaustive check, install testing dependencies and run tests as follows: (Linux)

pip install -r tools/ci_requirements.txt
export CI=‘true’
pytest tests/python/

4. Install the rest of the dependencies

requirements.txt

cd …/code/nerf/kaolin-wisp/

pip install -r requirements.txt -i https://pypi.douban.com/simple

5. Installing Wisp

python setup.py develop

Testing Your Installation

1. Download some sample data

RTMV dataset
Project: RTMV: A Ray-Traced Multi-View Synthetic Dataset for Novel View Synthesis

2. Train a NeRF using NGLOD

WISP_HEADLESS=1 python app/main.py --config configs/nglod_nerf.yaml --dataset-path /path/to/V8 --dataset-num-workers 4

3. Errors

AttributeError: partially initialized module ‘cv2’ has no attribute ‘_registerMatType’ (most likely due to a circular import)

安装低版本的opencv_python：
Reference：AttributeError: partially initialized module ‘cv2’ has no attribute ‘_registerMatType’

pip uninstall opencv-python
pip uninstall opencv-contrib-python
pip install opencv_python==3.4.10.37

TypeError: Descriptors cannot not be created directly.

安装低版本的protobuf:
Reference: TypeError: Descriptors cannot not be created directly.

pip uninstall protobuf
pip install protobuf==3.20.*