uzi007 commited on
Commit
2c7c419
1 Parent(s): 06bb0b4

Updated Dockerfile for GPU Dependencies

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -15,6 +15,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-4
15
  # Additional system dependencies
16
  RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp
17
 
 
 
18
  # Switch back to the root user to install Python packages
19
  USER root
20
 
 
15
  # Additional system dependencies
16
  RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp
17
 
18
+ FROM python:3.10
19
+
20
  # Switch back to the root user to install Python packages
21
  USER root
22