Spaces:
Build error
Build error
Update Dockerfile
Browse fileswip fix : lietorch : cuda not found
- 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 |
-
|
69 |
-
|
|
|
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/
|
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
|