Update Dockerfile

Dockerfile

@@ -47,16 +47,15 @@ ENV PYTHONUNBUFFERED=1
 
 # ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
 # ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
-# ENV FORCE_CUDA=1
+ENV FORCE_CUDA=1
 
 # Set the environment variable to specify the GPU device
-ENV CUDA_HOME=/bin/nvcc
+ENV CUDA_HOME=/usr/local/cuda
 # ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 # ENV CUDA_VISIBLE_DEVICES=0
 
-# ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
-# ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LD_LIBRARY_PATH}
-# ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:/usr/lib:/usr/lib64:${LIBRARY_PATH}
+ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/lib:/usr/lib64:${LD_LIBRARY_PATH}
 
 # Set home to the user's home directory
 ENV HOME=/home/user \
@@ -73,7 +72,7 @@ RUN pip install --upgrade pip ninja
 RUN pip install setuptools==69.5.1
 RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
 
-RUN python -c "import torch, os; print(torch.version.cuda); print(os.environ.get('CUDA_PATH')); print(os.environ.get('CUDA_HOME'))"
+RUN python -c "import torch, os; print(torch.version.cuda); print(os.environ.get('CUDA_HOME'))"
 COPY requirements.txt /tmp
 RUN cd /tmp && pip install -r requirements.txt