diff --git a/README.md b/README.md index 333796f4..76f5b3a0 100644 --- a/README.md +++ b/README.md @@ -64,11 +64,17 @@ ## Installation +SplaTAM has been tested on python 3.10, CUDA>=11.6. The simplest way to install all dependences is to use [anaconda](https://www.anaconda.com/) and [pip](https://pypi.org/project/pip/) in the following steps: -First, you have to make sure that you have all dependencies in place. -The simplest way is to use [anaconda](https://www.anaconda.com/). +```bash +conda create -n splatam python=3.10 +conda activate splatam +conda install -c "nvidia/label/cuda-11.6.0" cuda-toolkit +conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge +pip install -r requirements.txt +``` -You can create an anaconda environment called `splatam`. +We also provide a conda environment.yml file : ```bash conda env create -f environment.yml conda activate splatam diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..5594bbb4 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,15 @@ +tqdm==4.65.0 +Pillow +opencv-python +imageio +matplotlib +kornia +natsort +pyyaml +wandb +lpips +open3d==0.16.0 +torchmetrics +cyclonedds +pytorch-msssim +git+https://github.com/JonathonLuiten/diff-gaussian-rasterization-w-depth.git@cb65e4b86bc3bd8ed42174b72a62e8d3a3a71110 \ No newline at end of file