Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +4 -5
Dockerfile
CHANGED
|
@@ -13,17 +13,16 @@ ENV CUDA_HOME=/usr/local/cuda
|
|
| 13 |
# Set CUDA architecture flags
|
| 14 |
ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
# Clone the InstantSplat repository and download the pre-trained model
|
| 17 |
-
RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git
|
| 18 |
-
cd app && \
|
| 19 |
git submodule update --init --recursive && \
|
| 20 |
cd submodules/dust3r/ && \
|
| 21 |
mkdir -p checkpoints/ && \
|
| 22 |
wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/
|
| 23 |
|
| 24 |
-
RUN useradd -m -u 1000 user
|
| 25 |
-
WORKDIR /app
|
| 26 |
-
|
| 27 |
RUN pip install submodules/simple-knn && \
|
| 28 |
sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
|
| 29 |
pip install submodules/diff-gaussian-rasterization && \
|
|
|
|
| 13 |
# Set CUDA architecture flags
|
| 14 |
ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
|
| 15 |
|
| 16 |
+
RUN useradd -m -u 1000 user
|
| 17 |
+
WORKDIR /app
|
| 18 |
+
|
| 19 |
# Clone the InstantSplat repository and download the pre-trained model
|
| 20 |
+
RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git . && \
|
|
|
|
| 21 |
git submodule update --init --recursive && \
|
| 22 |
cd submodules/dust3r/ && \
|
| 23 |
mkdir -p checkpoints/ && \
|
| 24 |
wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/
|
| 25 |
|
|
|
|
|
|
|
|
|
|
| 26 |
RUN pip install submodules/simple-knn && \
|
| 27 |
sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
|
| 28 |
pip install submodules/diff-gaussian-rasterization && \
|