Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use NVIDIA CUDA as base for GPU support
-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+FROM ubuntu:22.04
 
 # Avoid prompts during installation
 ENV DEBIAN_FRONTEND=noninteractive
@@ -11,6 +11,8 @@ RUN apt-get update && apt-get install -y \
     python3 \
     python3-pip \
     wget \
+    docker.io \
+    sudo \
     && rm -rf /var/lib/apt/lists/*
 
 # Install code-server (VS Code Web)