Gregniuki commited on
Commit
a320acd
·
1 Parent(s): 5b0c0b4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -4,7 +4,8 @@ FROM debian:bullseye-slim
4
 
5
  # Install FFmpeg
6
  RUN apt-get update && \
7
- apt-get install -y python3-pip ffmpeg
 
8
 
9
  # Use the official Python 3.9 image as the base image
10
  #FROM python:3.9
 
4
 
5
  # Install FFmpeg
6
  RUN apt-get update && \
7
+ apt-get install -y python3-pip ffmpeg libcublas-dev
8
+ ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
9
 
10
  # Use the official Python 3.9 image as the base image
11
  #FROM python:3.9