SW777SW commited on
Commit
6593a75
·
verified ·
1 Parent(s): 2810f4b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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