Enable GPU with Docker

Dockerfile

@@ -1,19 +1,14 @@
-# Use NVIDIA image with PyTorch, CUDA, cuDNN preinstalled
 FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-runtime
 
-# Enable non-interactive install
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install dependencies
 RUN apt-get update && apt-get install -y \
     git ffmpeg libsndfile1 python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy and install Python dependencies
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
-# Copy app files
 COPY . /app
 WORKDIR /app