Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +8 -8
Dockerfile
CHANGED
@@ -45,14 +45,14 @@ FROM python:3.8
|
|
45 |
|
46 |
ENV PYTHONUNBUFFERED=1
|
47 |
|
48 |
-
ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
|
49 |
-
ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
|
50 |
-
ENV FORCE_CUDA=1
|
51 |
-
|
52 |
-
ENV CUDA_HOME=/usr/local/cuda
|
53 |
-
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
|
54 |
-
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LD_LIBRARY_PATH}
|
55 |
-
ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LIBRARY_PATH}
|
56 |
|
57 |
# Set home to the user's home directory
|
58 |
ENV HOME=/home/user \
|
|
|
45 |
|
46 |
ENV PYTHONUNBUFFERED=1
|
47 |
|
48 |
+
# ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
|
49 |
+
# ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
|
50 |
+
# ENV FORCE_CUDA=1
|
51 |
+
|
52 |
+
# ENV CUDA_HOME=/usr/local/cuda
|
53 |
+
# ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
|
54 |
+
# ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LD_LIBRARY_PATH}
|
55 |
+
# ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LIBRARY_PATH}
|
56 |
|
57 |
# Set home to the user's home directory
|
58 |
ENV HOME=/home/user \
|