uzi007 commited on
Commit
06bb0b4
·
1 Parent(s): c82d6c0

Updated Dockerfile for GPU Dependencies

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -6,6 +6,9 @@ ENV CUDA_HOME=/usr/local/cuda
6
  ENV PATH=$CUDA_HOME/bin:$PATH
7
  ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
8
 
 
 
 
9
  # Install NVIDIA driver
10
  RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
11
 
 
6
  ENV PATH=$CUDA_HOME/bin:$PATH
7
  ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
8
 
9
+ # Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts
10
+ ENV DEBIAN_FRONTEND=noninteractive
11
+
12
  # Install NVIDIA driver
13
  RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
14