Fix environment to work on AI machine (#1)

* Fix environment to work on AI machine

* move simple-knn install inside docker container

* fix

Dockerfile

@@ -63,4 +63,3 @@ RUN cd .. && \
     cmake .. -GNinja && \
     ninja && \
     ninja install
-

README.md

@@ -39,13 +39,14 @@ For development, we suggest running the Docker conrtainer in interactive mode.
 To do so, you must mount the proper resources. 
 
 ```shell
-nvidia-docker run -it --rm -v $(pwd):/workspace/ gaussian_splatting 
+docker run -it --gpus all --rm -v $(pwd):/workspace/ gaussian_splatting 
 ```
 
 ### 4. Activate the Conda Environment.
 
 ```shell
-conda activate gaussian_activate
+conda activate gaussian_splatting
+pip install submodules/diff-gaussian-rasterization submodules/simple-knn
 ```
 
 ### 5. [Optional] Test your Environment. 
@@ -131,14 +132,6 @@ camera poses during training, pass the `--pose-free` flag to training script.
 python scripts/train.py -s <path to dataset>
 ```
 
-### 3. Evaluation
-
-```shell
-python scripts/render.py -m <path to trained model>
-python scripts/metrics.py -m <path to trained model> 
-```
-
-
 # Viewer
 
 To view the resulting model, we recommend using this web viewer:

environment.yml

@@ -1,17 +1,295 @@
 name: gaussian_splatting
 channels:
   - pytorch
+  - nvidia
   - conda-forge
   - defaults
 dependencies:
-  - plyfile=0.8.1
-  - python=3.7.13
-  - pip=22.3.1
-  - opencv 
-  - pytorch=1.12.1
-  - torchvision=0.13.1
-  - tqdm
+  - _libgcc_mutex=0.1=conda_forge
+  - _openmp_mutex=4.5=2_gnu
+  - alsa-lib=1.2.8=h166bdaf_0
+  - aom=3.5.0=h27087fc_0
+  - attr=2.5.1=h166bdaf_1
+  - blas=1.0=mkl
+  - brotli-python=1.0.9=py38h6a678d5_8
+  - bzip2=1.0.8=hd590300_5
+  - c-ares=1.28.1=hd590300_0
+  - ca-certificates=2024.3.11=h06a4308_0
+  - cairo=1.16.0=ha61ee94_1014
+  - certifi=2024.2.2=py38h06a4308_0
+  - colorama=0.4.6=pyhd8ed1ab_0
+  - cuda=11.6.1=0
+  - cuda-cccl=11.6.55=hf6102b2_0
+  - cuda-command-line-tools=11.6.2=0
+  - cuda-compiler=11.6.2=0
+  - cuda-cudart=11.6.55=he381448_0
+  - cuda-cudart-dev=11.6.55=h42ad0f4_0
+  - cuda-cuobjdump=11.6.124=h2eeebcb_0
+  - cuda-cupti=11.6.124=h86345e5_0
+  - cuda-cuxxfilt=11.6.124=hecbf4f6_0
+  - cuda-driver-dev=11.6.55=0
+  - cuda-gdb=12.4.127=0
+  - cuda-libraries=11.6.1=0
+  - cuda-libraries-dev=11.6.1=0
+  - cuda-memcheck=11.8.86=0
+  - cuda-nsight=12.4.127=0
+  - cuda-nsight-compute=12.4.1=0
+  - cuda-nvcc=11.6.124=hbba6d2d_0
+  - cuda-nvdisasm=12.4.127=0
+  - cuda-nvml-dev=11.6.55=haa9ef22_0
+  - cuda-nvprof=12.4.127=0
+  - cuda-nvprune=11.6.124=he22ec0a_0
+  - cuda-nvrtc=11.6.124=h020bade_0
+  - cuda-nvrtc-dev=11.6.124=h249d397_0
+  - cuda-nvtx=11.6.124=h0630a44_0
+  - cuda-nvvp=12.4.127=0
+  - cuda-runtime=11.6.1=0
+  - cuda-samples=11.6.101=h8efea70_0
+  - cuda-sanitizer-api=12.4.127=0
+  - cuda-toolkit=11.6.1=0
+  - cuda-tools=11.6.1=0
+  - cuda-visual-tools=11.6.1=0
+  - dav1d=1.2.1=hd590300_0
+  - dbus=1.13.6=h5008d03_3
+  - double-conversion=3.3.0=h59595ed_0
+  - expat=2.6.2=h59595ed_0
+  - ffmpeg=4.4.2=gpl_h8dda1f0_112
+  - fftw=3.3.10=nompi_hc118613_108
+  - font-ttf-dejavu-sans-mono=2.37=hab24e00_0
+  - font-ttf-inconsolata=3.000=h77eed37_0
+  - font-ttf-source-code-pro=2.038=h77eed37_0
+  - font-ttf-ubuntu=0.83=h77eed37_2
+  - fontconfig=2.14.2=h14ed4e7_0
+  - fonts-conda-ecosystem=1=0
+  - fonts-conda-forge=1=0
+  - freeglut=3.2.2=h9c3ff4c_1
+  - freetype=2.12.1=h267a509_2
+  - fribidi=1.0.10=h36c2ea0_0
+  - gds-tools=1.9.1.3=0
+  - gettext=0.22.5=h59595ed_2
+  - gettext-tools=0.22.5=h59595ed_2
+  - glib=2.80.0=hf2295e7_6
+  - glib-tools=2.80.0=hde27a5a_6
+  - gmp=6.3.0=h59595ed_1
+  - gnutls=3.7.9=hb077bed_0
+  - graphite2=1.3.13=h59595ed_1003
+  - gst-plugins-base=1.21.3=h4243ec0_1
+  - gstreamer=1.21.3=h25f0c4b_1
+  - gstreamer-orc=0.4.38=hd590300_0
+  - harfbuzz=4.4.1=hf9f4e7c_0
+  - hdf5=1.12.1=h2b7332f_3
+  - icu=70.1=h27087fc_0
+  - idna=3.7=py38h06a4308_0
+  - imath=3.1.11=hfc55251_0
+  - intel-openmp=2022.1.0=h9e868ea_3769
+  - jack=1.9.22=h11f4161_0
+  - jasper=2.0.33=h0ff4b12_1
+  - jpeg=9e=h5eee18b_1
+  - keyutils=1.6.1=h166bdaf_0
+  - krb5=1.20.1=h81ceb04_0
+  - lame=3.100=h166bdaf_1003
+  - lcms2=2.12=h3be6417_0
+  - ld_impl_linux-64=2.40=h55db66e_0
+  - lerc=4.0.0=h27087fc_0
+  - libabseil=20240116.2=cxx17_h59595ed_0
+  - libaec=1.1.3=h59595ed_0
+  - libasprintf=0.22.5=h661eb56_2
+  - libasprintf-devel=0.22.5=h661eb56_2
+  - libass=0.14.0=hc2a3f6d_3
+  - libblas=3.9.0=16_linux64_mkl
+  - libcap=2.66=ha37c62d_0
+  - libcblas=3.9.0=16_linux64_mkl
+  - libclang=15.0.7=default_h127d8a8_5
+  - libclang13=15.0.7=default_h5d6823c_5
+  - libcublas=11.9.2.110=h5e84587_0
+  - libcublas-dev=11.9.2.110=h5c901ab_0
+  - libcufft=10.7.1.112=hf425ae0_0
+  - libcufft-dev=10.7.1.112=ha5ce4c0_0
+  - libcufile=1.9.1.3=0
+  - libcufile-dev=1.9.1.3=0
+  - libcups=2.3.3=h36d4200_3
+  - libcurand=10.3.5.147=0
+  - libcurand-dev=10.3.5.147=0
+  - libcurl=8.5.0=h251f7ec_0
+  - libcusolver=11.3.4.124=h33c3c4e_0
+  - libcusparse=11.7.2.124=h7538f96_0
+  - libcusparse-dev=11.7.2.124=hbbe9722_0
+  - libdb=6.2.32=h6a678d5_1
+  - libdeflate=1.14=h166bdaf_0
+  - libdrm=2.4.120=hd590300_0
+  - libedit=3.1.20191231=he28a2e2_2
+  - libev=4.33=hd590300_2
+  - libevent=2.1.10=h28343ad_4
+  - libexpat=2.6.2=h59595ed_0
+  - libffi=3.4.2=h7f98852_5
+  - libflac=1.4.3=h59595ed_0
+  - libgcc-ng=13.2.0=h77fa898_6
+  - libgcrypt=1.10.3=hd590300_0
+  - libgettextpo=0.22.5=h59595ed_2
+  - libgettextpo-devel=0.22.5=h59595ed_2
+  - libgfortran-ng=13.2.0=h69a702a_6
+  - libgfortran5=13.2.0=h43f5ff8_6
+  - libglib=2.80.0=hf2295e7_6
+  - libglu=9.0.0=hf484d3e_1
+  - libgomp=13.2.0=h77fa898_6
+  - libgpg-error=1.49=h4f305b6_0
+  - libhwloc=2.9.1=hd6dc26d_0
+  - libiconv=1.17=hd590300_2
+  - libidn2=2.3.7=hd590300_0
+  - libjpeg-turbo=2.0.0=h9bf148f_0
+  - liblapack=3.9.0=16_linux64_mkl
+  - liblapacke=3.9.0=16_linux64_mkl
+  - libllvm15=15.0.7=hadd5161_1
+  - libnghttp2=1.57.0=h2d74bed_0
+  - libnpp=11.6.3.124=hd2722f0_0
+  - libnpp-dev=11.6.3.124=h3c42840_0
+  - libnsl=2.0.1=hd590300_0
+  - libnvjpeg=11.6.2.124=hd473ad6_0
+  - libnvjpeg-dev=11.6.2.124=hb5906b9_0
+  - libogg=1.3.4=h7f98852_1
+  - libopenblas=0.3.27=pthreads_h413a1c8_0
+  - libopencv=4.6.0=py38hc65905f_1
+  - libopenvino=2023.2.0=h59595ed_0
+  - libopenvino-auto-batch-plugin=2023.2.0=h59595ed_0
+  - libopenvino-auto-plugin=2023.2.0=h59595ed_0
+  - libopenvino-hetero-plugin=2023.2.0=h59595ed_0
+  - libopenvino-intel-cpu-plugin=2023.2.0=h59595ed_0
+  - libopenvino-intel-gpu-plugin=2023.2.0=h59595ed_0
+  - libopenvino-ir-frontend=2023.2.0=h59595ed_0
+  - libopenvino-onnx-frontend=2023.2.0=h59595ed_0
+  - libopenvino-paddle-frontend=2023.2.0=h59595ed_0
+  - libopenvino-pytorch-frontend=2023.2.0=h59595ed_0
+  - libopenvino-tensorflow-frontend=2023.2.0=h59595ed_0
+  - libopenvino-tensorflow-lite-frontend=2023.2.0=h59595ed_0
+  - libopus=1.3.1=h7f98852_1
+  - libpciaccess=0.18=hd590300_0
+  - libpng=1.6.43=h2797004_0
+  - libpq=15.2=hb675445_0
+  - libprotobuf=3.20.3=he621ea3_0
+  - libsndfile=1.2.2=hc60ed4a_1
+  - libsqlite=3.45.3=h2797004_0
+  - libssh2=1.10.0=hdbd6064_3
+  - libstdcxx-ng=13.2.0=hc0a3c3a_6
+  - libsystemd0=252=h2a991cd_0
+  - libtasn1=4.19.0=h166bdaf_0
+  - libtiff=4.4.0=h82bc61c_5
+  - libtool=2.4.7=h27087fc_0
+  - libudev1=253=h0b41bf4_0
+  - libunistring=0.9.10=h7f98852_0
+  - libuuid=2.38.1=h0b41bf4_0
+  - libva=2.18.0=h0b41bf4_0
+  - libvorbis=1.3.7=h9c3ff4c_0
+  - libvpx=1.11.0=h295c915_0
+  - libwebp-base=1.4.0=hd590300_0
+  - libxcb=1.13=h1bed415_1
+  - libxcrypt=4.4.36=hd590300_1
+  - libxkbcommon=1.5.0=h79f4944_1
+  - libxml2=2.10.3=hca2bb57_4
+  - libzlib=1.2.13=hd590300_5
+  - lz4-c=1.9.4=hcb278e6_0
+  - mkl=2022.1.0=hc2b9512_224
+  - mpg123=1.32.6=h59595ed_0
+  - mysql-common=8.0.32=ha901b37_0
+  - mysql-libs=8.0.32=hd7da12d_0
+  - ncurses=6.4.20240210=h59595ed_0
+  - nettle=3.9.1=h7ab15ed_0
+  - nsight-compute=2024.1.1.4=0
+  - nspr=4.35=h6a678d5_0
+  - nss=3.89.1=h6a678d5_0
+  - numpy=1.24.4=py38h59b608b_0
+  - ocl-icd=2.3.2=hd590300_1
+  - ocl-icd-system=1.0.0=1
+  - opencv=4.6.0=py38h578d9bd_1
+  - openexr=3.2.2=haf962dd_1
+  - openh264=2.3.1=hcb278e6_2
+  - openjpeg=2.4.0=h3ad879b_0
+  - openssl=3.0.13=h7f8727e_1
+  - p11-kit=0.24.1=hc5aa10d_0
+  - pcre2=10.43=hcad00b1_0
+  - pillow=10.3.0=py38h5eee18b_0
+  - pip=22.3.1=pyhd8ed1ab_0
+  - pixman=0.43.2=h59595ed_0
+  - plyfile=0.8.1=pyhd8ed1ab_0
+  - pthread-stubs=0.4=h36c2ea0_1001
+  - pugixml=1.14=h59595ed_0
+  - pulseaudio=16.1=ha8d29e2_1
+  - py-opencv=4.6.0=py38h7f3c49e_1
+  - pysocks=1.7.1=py38h06a4308_0
+  - python=3.8.16=he550d4f_1_cpython
+  - python_abi=3.8=4_cp38
+  - pytorch=1.13.1=py3.8_cuda11.6_cudnn8.3.2_0
+  - pytorch-cuda=11.6=h867d48c_1
+  - pytorch-mutex=1.0=cuda
+  - qt-main=5.15.6=hf6cd601_5
+  - readline=8.2=h8228510_1
+  - requests=2.31.0=py38h06a4308_1
+  - setuptools=69.5.1=pyhd8ed1ab_0
+  - snappy=1.1.10=h6a678d5_1
+  - sqlite=3.45.3=h5eee18b_0
+  - svt-av1=1.4.1=hcb278e6_0
+  - tbb=2021.8.0=hdb19cb5_0
+  - tk=8.6.13=noxft_h4845f30_101
+  - torchvision=0.14.1=py38_cu116
+  - tqdm=4.66.2=pyhd8ed1ab_0
+  - typing_extensions=4.9.0=py38h06a4308_1
+  - wheel=0.43.0=pyhd8ed1ab_1
+  - x264=1!164.3095=h166bdaf_2
+  - x265=3.5=h924138e_3
+  - xcb-util=0.4.0=h516909a_0
+  - xcb-util-cursor=0.1.4=hd590300_0
+  - xcb-util-image=0.4.0=h166bdaf_0
+  - xcb-util-keysyms=0.4.0=h516909a_0
+  - xcb-util-renderutil=0.3.9=h166bdaf_0
+  - xcb-util-wm=0.4.1=h516909a_0
+  - xkeyboard-config=2.38=h0b41bf4_0
+  - xorg-fixesproto=5.0=h7f98852_1002
+  - xorg-inputproto=2.3.2=h7f98852_1002
+  - xorg-kbproto=1.0.7=h7f98852_1002
+  - xorg-libice=1.1.1=hd590300_0
+  - xorg-libsm=1.2.4=h7391055_0
+  - xorg-libx11=1.8.4=h0b41bf4_0
+  - xorg-libxau=1.0.11=hd590300_0
+  - xorg-libxdmcp=1.1.3=h7f98852_0
+  - xorg-libxext=1.3.4=h0b41bf4_2
+  - xorg-libxfixes=5.0.3=h7f98852_1004
+  - xorg-libxi=1.7.10=h7f98852_0
+  - xorg-libxrender=0.9.10=h7f98852_1003
+  - xorg-renderproto=0.11.1=h7f98852_1002
+  - xorg-xextproto=7.3.0=h0b41bf4_1003
+  - xorg-xproto=7.0.31=h7f98852_1007
+  - xz=5.2.6=h166bdaf_0
+  - zlib=1.2.13=hd590300_5
+  - zstd=1.5.5=hfc55251_0
   - pip:
-    - -e .
-    - submodules/diff-gaussian-rasterization
-    - submodules/simple-knn
+      - -e .
+      - black==24.4.2
+      - chardet==5.2.0
+      - charset-normalizer==3.3.2
+      - click==8.1.7
+      - contourpy==1.1.1
+      - cycler==0.12.1
+      - filelock==3.14.0
+      - fonttools==4.51.0
+      - fsspec==2024.3.1
+      - huggingface-hub==0.23.0
+      - importchecker==3.0
+      - importlib-resources==6.4.0
+      - isort==5.13.2
+      - kiwisolver==1.4.5
+      - matplotlib==3.7.5
+      - mypy-extensions==1.0.0
+      - packaging==24.0
+      - pathspec==0.12.1
+      - platformdirs==4.2.1
+      - pyparsing==3.1.2
+      - python-dateutil==2.9.0.post0
+      - pyyaml==6.0.1
+      - regex==2024.4.28
+      - safetensors==0.4.3
+      - six==1.16.0
+      - tokenizers==0.19.1
+      - tomli==2.0.1
+      - transformers==4.40.1
+      - typing-extensions==4.11.0
+      - urllib3==2.2.1
+      - zipp==3.18.1

setup.py

@@ -11,6 +11,7 @@
         "isort",
         "importchecker",
         "matplotlib",
-        "transformers"
+        "transformers",
+        "chardet"
     ]
 )