Skip to content

Ubuntu PyTorch CUDA Docker image with KDE Plasma Desktop & VNC. Ideal for LLM & Deep Learning remote work.

License

Notifications You must be signed in to change notification settings

ivangabriele/docker-cuda-desktop

Docker Image :: Ubuntu KDE Plasma Desktop with PyTorch, CUDA & VNC

img-github img-docker img-runpod

Ideal for LLM & Deep Learning remote work.

Screenshot

CUDA Desktop

Content

Base

  • PyTorch (base image including Ubuntu & CUDA)
  • KDE Plasma Desktop
  • VNC Server (port 5900)

Main Applications

  • Firefox (ESR)
  • Kitty (terminal)
  • Sublime Text
  • Visual Studio Code (must be run with --no-sandbox flag)

Nice to Have

  • Fira Code (monospace font)
  • zsh (with oh-my-zsh)

Tags

Pytorch tag as per their official documentation:

  • PyTorch 23.09-py3
    • latest

Environment Variables

  • VNC_PASSWORD: Password for VNC connection. Default: password.
  • VNC_SCREEN_SIZE: Screen size. Default: 1920x1080.

Usage

To connect to the container's desktop, you can use any VNC client.

Note

You will have a wide list of errors during startup but you can ignore them. Enulating a full X server in Docker is not an easy task.

Your server will be ready once you see this line in the logs:

[...]
cuda-desktop-server  | Plasma Shell startup completed
[...]

Deployment

RunPod

img-runpod

Note

You can create and attach a Storage Network Volume to your RunPod pod to automatically sync and persist your /home/ubuntu directory between deployments.

Backup Volume in Deployment

When deploying your container, you can also attach a shared storage as a volume for backup.

Simply mount your desired backup directory to /volumes/home_ubuntu_backup in your Docker run command or Docker composer file. It will be automatically retored (on subsequent boots) and then synced each minute.

Thanks


About

Ubuntu PyTorch CUDA Docker image with KDE Plasma Desktop & VNC. Ideal for LLM & Deep Learning remote work.

Topics

Resources

License

Code of conduct

Contributing

Security policy

Stars

Watchers

Forks