radames commited on
Commit
04b0278
·
1 Parent(s): 10b593d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -52,7 +52,7 @@ RUN pip install --upgrade pip setuptools ninja
52
  RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118
53
  # Install nerfacc and tiny-cuda-nn before installing requirements.txt
54
  # because these two installations are time consuming and error prone
55
- RUN pip install git+https://github.com/KAIR-BAIR/nerfacc.git@v0.5.2
56
  RUN pip install git+https://github.com/NVlabs/tiny-cuda-nn.git#subdirectory=bindings/torch
57
 
58
  COPY requirements.txt /tmp
 
52
  RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118
53
  # Install nerfacc and tiny-cuda-nn before installing requirements.txt
54
  # because these two installations are time consuming and error prone
55
+ RUN pip install nerfacc==0.5.2 -f https://nerfacc-bucket.s3.us-west-2.amazonaws.com/whl/torch-2.0.0_cu118.html
56
  RUN pip install git+https://github.com/NVlabs/tiny-cuda-nn.git#subdirectory=bindings/torch
57
 
58
  COPY requirements.txt /tmp