Gregniuki commited on
Commit
e4345d4
·
1 Parent(s): 15063c3

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -64,7 +64,7 @@ RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/
64
 
65
  # Install FFmpeg
66
  RUN apt-get update && \
67
- apt-get install -y python3-pip ffmpeg cuda
68
 
69
 
70
  # Use the official Python 3.9 image as the base image
 
64
 
65
  # Install FFmpeg
66
  RUN apt-get update && \
67
+ apt-get install -y python3-pip ffmpeg libcublas-11-8
68
 
69
 
70
  # Use the official Python 3.9 image as the base image