Update Dockerfile

Dockerfile

@@ -1,5 +1,10 @@
-# Use the official Visual Studio Code Server image
-FROM codercom/code-server:latest
+# Use NVIDIA CUDA image
+FROM nvidia/cuda:12.9.1-cudnn-devel-ubuntu24.04
+
+RUN apt install curl wget dumb-init -y
+
+RUN curl http://code-server.dev/install.sh -fsSL | bash
+
 # Set the user as root with the custom name
 USER root
 # Expose the default port used by Visual Studio Code Server