tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-12 10:55:57 +08:00
Code Activity

Upgrade pip and install 'wheel' in Dockerfile

Browse Source 
Added installation of 'wheel' package and upgraded 'setuptools' in Dockerfile.
This commit is contained in:
basebala authored and GitHub committed 2026-07-08 10:08:28 -07:00
1 parent 61b83576b4
commit 32be3d3244
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+2 -1
View File
@@ -81,10 +81,11 @@ RUN git clone https://github.com/colmap/colmap.git && \
# Upgrade pip and install dependencies.
# pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118 && \
RUN pip install --no-cache-dir --upgrade pip 'setuptools<70.0.0' && \
RUN pip install --no-cache-dir --upgrade pip 'setuptools<70.0.0' 'wheel' && \
pip install --no-cache-dir torch==2.1.2+cu118 torchvision==0.16.2+cu118 'numpy<2.0.0' --extra-index-url https://download.pytorch.org/whl/cu118 && \
git clone --branch master --recursive https://github.com/cvg/Hierarchical-Localization.git /opt/hloc && \
cd /opt/hloc && git checkout v1.4 && python3.10 -m pip install --no-cache-dir . && cd ~ && \
pip install --no-cache-dir --upgrade 'setuptools<70.0.0' 'wheel' && \
TCNN_CUDA_ARCHITECTURES="${CUDA_ARCHITECTURES}" pip install --no-cache-dir "git+https://github.com/NVlabs/tiny-cuda-nn.git@b3473c81396fe927293bdfd5a6be32df8769927c#subdirectory=bindings/torch" && \
pip install --no-cache-dir pycolmap==0.6.1 pyceres==2.1 omegaconf==2.3.0