Create Dockerfile

Dockerfile

@@ -0,0 +1,20 @@
+# Using Ubuntu as the base image
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+
+# Setting environment variables
+ENV TMATE_SOCK=/tmp/tmate.sock
+
+# Installing necessary packages
+RUN apt-get update && \
+    apt-get install -y sudo curl unzip jq xrdp kubuntu-desktop tmate
+
+# Creating a directory for the script
+WORKDIR /app
+
+# Copying the entrypoint script to the container
+COPY entrypoint.sh /app
+
+# Making the entrypoint script executable
+RUN chmod +x /app/entrypoint.sh
+
+CMD ["/app/entrypoint.sh"]