this is a docker image of ZipNeRF + CamP. https://github.com/jonbarron/camp_zipnerf
69
Dockerfile
FROM nvcr.io/nvidia/jax:23.10-py3
RUN pip install --no-cache-dir \
git+https://github.com/rmbrualla/pycolmap.git \
dm-pix==0.4.2 \
ffmpeg \
gin-config==0.5.0 \
immutabledict==4.1.0 \
jaxcam==0.1.1 \
mediapy==1.2.0 \
ml_collections \
opencv-python-headless \
pillow==10.2.0 \
rawpy==0.19.0 \
tensorboard==2.16.1 \
tensorflow==2.16.1
# matplotlib version is not mentioned. but a source of bug.
RUN pip install --no-cache-dir \
matplotlib==3.7.0
All jax-related dependencies are handled by the upstream Nvidia jax image. I am not pip installing them again.
The jax version in nvidia/jax:23.10-py3 is slightly older than camp_zipnerf's requirement, but in practice it's fine.
To run camp_zipnerf, you need to clone the official repo. Rmb to delete all the dependencies in the repo's toml file, before pip install -e .. We don't want the container's built-in jax to be uninstalled.
Content type
Image
Digest
sha256:67a65109c…
Size
6.9 GB
Last updated
over 1 year ago
docker pull haochenw/camp_zipnerf:250226