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 \
|