gaussian-splatting/docker/initialize.sh

43 lines
1.1 KiB
Bash
Raw Normal View History

2023-10-25 13:24:31 +00:00
#!/bin/bash
2023-10-27 09:54:26 +00:00
#git clone https://github.com/NVIDIA/cuda-samples #<- To Test docker Cuda image..
2023-10-25 15:57:21 +00:00
#cd cuda-samples
2023-10-25 13:24:31 +00:00
git clone https://ceres-solver.googlesource.com/ceres-solver
cd ceres-solver
git checkout $(git describe --tags) # Checkout the latest release
mkdir build
cd build
2023-10-25 17:15:14 +00:00
cmake .. -DBUILD_TESTING=OFF -DBUILD_EXAMPLES=OFF -DCMAKE_CUDA_ARCHITECTURES="60;70;80" #-DUSE_CUDA=OFF
2023-10-25 13:24:31 +00:00
make
sudo make install
cd ..
cd ..
git clone https://github.com/colmap/colmap
cd colmap
git checkout dev
mkdir build
cd build
2023-10-25 17:15:14 +00:00
cmake .. -DCMAKE_CUDA_ARCHITECTURES="60;70;80" #-DCUDA_ENABLED=OFF
2023-10-25 13:24:31 +00:00
make
sudo make install
cd ..
cd ..
2023-10-28 14:56:50 +00:00
sudo apt-get -y install cuda
2023-10-28 15:35:50 +00:00
sudo apt -y install nvidia-cuda-toolkit
2023-10-28 14:56:50 +00:00
2023-10-28 15:35:50 +00:00
python3.10 -m pip install plyfile tqdm
2023-10-28 14:56:50 +00:00
python3.10 -m pip install https://huggingface.co/camenduru/gaussian-splatting/resolve/main/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
python3.10 -m pip install https://huggingface.co/camenduru/gaussian-splatting/resolve/main/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl
2023-10-27 09:54:26 +00:00
python3.10 -m pip install torchvision
2023-10-25 17:15:14 +00:00
ln -s docker/run.sh ./run.sh
2023-10-25 13:24:31 +00:00
exit 0