cuda added

Dockerfile

@@ -1,14 +1,17 @@
-FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-runtime
+# Use an NVIDIA PyTorch container with cuDNN 9.1 support
+FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
 
+# Basic setup
 ENV DEBIAN_FRONTEND=noninteractive
-
 RUN apt-get update && apt-get install -y \
-    git ffmpeg libsndfile1 python3-dev \
+    python3 python3-pip ffmpeg git libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
+# Set up Python environment
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
+# Copy app files
 COPY . /app
 WORKDIR /app
 

backend/services/interview_engine.py

@@ -15,6 +15,11 @@ if torch.cuda.is_available():
     print("cuDNN version:", torch.backends.cudnn.version())
 else:
     print("❌ CUDA Not Available")
+print("🔥 CUDA:", torch.cuda.is_available())
+print("🧠 GPU:", torch.cuda.get_device_name(0))
+print("💡 cuDNN version:", torch.backends.cudnn.version())
+print("💥 cuDNN enabled:", torch.backends.cudnn.is_available())
+
 
 
 # Initialize models