Mightypeacock commited on
Commit
5a76e6b
·
verified ·
1 Parent(s): 8a107a6

Update Dockerfile

Browse files

wip fix : lietorch : cuda not found

Files changed (1) hide show
  1. Dockerfile +6 -6
Dockerfile CHANGED
@@ -63,18 +63,18 @@ RUN ls -la /usr/local/cuda/bin/nvcc
63
 
64
  # 5. Install with explicit CUDA flags
65
  RUN cd thirdparty/lietorch && \
66
- export TORCH_CUDA_ARCH_LIST="8.6" && \
67
  export FORCE_CUDA=1 && \
68
- pip install --no-build-isolation -v -e . && \
69
- grep "Successfully installed" install.log && \
 
70
  cd /app
71
 
72
 
73
- RUN pip install -v --no-cache-dir --no-build-isolation -e thirdparty/lietorch/
74
 
75
  # Install custom packages from the cloned repository
76
- RUN python -m pip install -e thirdparty/lietorch/ && \
77
- python -m pip install -e thirdparty/diff-gaussian-rasterization-w-pose/ && \
78
  python -m pip install -e thirdparty/simple-knn/
79
 
80
  # Install the main package
 
63
 
64
  # 5. Install with explicit CUDA flags
65
  RUN cd thirdparty/lietorch && \
66
+ export TORCH_CUDA_ARCH_LIST="8.6" && \
67
  export FORCE_CUDA=1 && \
68
+ export CUDA_HOME=/usr/local/cuda && \
69
+ export PATH="${CUDA_HOME}/bin:${PATH}" && \
70
+ pip install --no-build-isolation --config-settings editable_mode=compat -v -e . && \
71
  cd /app
72
 
73
 
74
+ #RUN pip install -v --no-cache-dir --no-build-isolation -e thirdparty/lietorch/
75
 
76
  # Install custom packages from the cloned repository
77
+ RUN python -m pip install -e thirdparty/diff-gaussian-rasterization-w-pose/ && \
 
78
  python -m pip install -e thirdparty/simple-knn/
79
 
80
  # Install the main package