Yuliang commited on
Commit
9e5f76e
1 Parent(s): aa66fcc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 \