installing dependencies on CUDA image

Dockerfile

@@ -5,6 +5,14 @@ FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 
+RUN apt-get update && apt-get install --no-install-recommends -y \
+  build-essential \
+  python3.9 \
+  python3-pip \
+  git \
+  ffmpeg \
+  && apt-get clean && rm -rf /var/lib/apt/lists/*
+
 # Set environment variables to prevent Python from writing pyc files to disk
 # and to force unbuffered output (useful for Docker)
 ENV PYTHONDONTWRITEBYTECODE 1