Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +6 -4
Dockerfile
CHANGED
|
@@ -13,7 +13,7 @@ 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 |
-
RUN useradd -m -u 1000 user
|
| 17 |
WORKDIR /app
|
| 18 |
|
| 19 |
# Clone the InstantSplat repository and download the pre-trained model
|
|
@@ -44,13 +44,15 @@ RUN pip install submodules/simple-knn && \
|
|
| 44 |
|
| 45 |
# Set the working directory
|
| 46 |
|
| 47 |
-
COPY --chown=user ./requirements.txt requirements.txt
|
|
|
|
| 48 |
RUN pip install -r requirements.txt
|
| 49 |
|
| 50 |
-
USER user
|
| 51 |
ENV PATH="/home/user/.local/bin:$PATH"
|
| 52 |
|
| 53 |
-
COPY --chown=user ./run_train_infer.sh run_train_infer.sh
|
|
|
|
| 54 |
RUN chmod +x run_train_infer.sh
|
| 55 |
|
| 56 |
|
|
|
|
| 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
|
|
|
|
| 44 |
|
| 45 |
# Set the working directory
|
| 46 |
|
| 47 |
+
# COPY --chown=user ./requirements.txt requirements.txt
|
| 48 |
+
COPY ./requirements.txt requirements.txt
|
| 49 |
RUN pip install -r requirements.txt
|
| 50 |
|
| 51 |
+
# USER user
|
| 52 |
ENV PATH="/home/user/.local/bin:$PATH"
|
| 53 |
|
| 54 |
+
# COPY --chown=user ./run_train_infer.sh run_train_infer.sh
|
| 55 |
+
COPY ./run_train_infer.sh run_train_infer.sh
|
| 56 |
RUN chmod +x run_train_infer.sh
|
| 57 |
|
| 58 |
|