Dockerfile

# 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"]