Update Dockerfile

Dockerfile

@@ -49,7 +49,11 @@ ENV PYTHONUNBUFFERED=1
 # ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
 # ENV FORCE_CUDA=1
 
-ENV CUDA_HOME=/usr/local/cuda
+# Set the environment variable to specify the GPU device
+ENV CUDA_HOME=/usr/local/cuda:/usr/local:/usr/bin
+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}