Dockerfile

FROM dromni/nerfstudio:0.3.2

ARG CUDA_VERSION=11.8.0
ARG CUDA_ARCHITECTURES=86
ARG OS_VERSION=22.04

RUN ["ns-download-data", "nerfstudio", "--capture-name=poster"]

CMD ["ns-train", "nerfacto", "--data", "data/nerfstudio/poster"]