uzi007 commited on
Commit
c82d6c0
·
1 Parent(s): 29e79f8

Updated Dockerfile for GPU Dependencies

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -7,7 +7,7 @@ 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.106
11
 
12
  # Additional system dependencies
13
  RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp
 
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
 
12
  # Additional system dependencies
13
  RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp