From 90ddaf2f9632ff373acf8ca41bf4b33d3ef232b6 Mon Sep 17 00:00:00 2001 From: mgrankin Date: Thu, 16 Apr 2020 16:38:45 +0300 Subject: [PATCH 1/2] conda environment --- README.md | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 67a04c3..0992537 100644 --- a/README.md +++ b/README.md @@ -61,13 +61,23 @@ Optimizing a NeRF takes between a few hours and a day or two (depending on resol ## How to train your NeRF super-quickly! -To train a "full" NeRF model (i.e., using 3D coordinates as well as ray directions, and the hierarchical sampling procedure), first setup dependencies. In a new `conda` or `virtualenv` environment, run +To train a "full" NeRF model (i.e., using 3D coordinates as well as ray directions, and the hierarchical sampling procedure), first setup dependencies. + +### Using pip +In a new `conda` or `virtualenv` environment, run +using pip ```bash pip install -r requirements.txt ``` - **Importantly**, install [torchsearchsorted](https://github.com/aliutkus/torchsearchsorted) by following instructions from their `README`. +### Alternatively, using conda +```bash +cd nerf-pytorch +conda env create +conda activate nerf +``` + Once everything is setup, to run experiments, first edit `config/lego.yml` to specify your own parameters. The training script can be invoked by running From 54788162f1bbe5ea0f40a26d3de9124fb555750f Mon Sep 17 00:00:00 2001 From: mgrankin Date: Thu, 16 Apr 2020 16:54:25 +0300 Subject: [PATCH 2/2] environment --- environment.yml | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 environment.yml diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..ae9dde3 --- /dev/null +++ b/environment.yml @@ -0,0 +1,21 @@ +name: nerf +channels: + - pytorch + - conda-forge + - defaults +dependencies: + - python>=3.7 + - pip + - pytorch>=1.3 + - cuda100 + - cudatoolkit>=10.1 + - cudatoolkit-dev + - imageio + - numpy + - pyyaml + - tensorboard + - torchvision + - tqdm + - pip: + - opencv-python-headless + - "git+https://github.com/aliutkus/torchsearchsorted#egg=torchsearchsorted"