Skip to Content
GuidesUbuntu GPU Workstation

Ubuntu GPU Workstation

Build an Ubuntu workstation with NVIDIA GPU support for machine learning.

Prompt

Create an Ubuntu 24.04 GPU workstation for machine learning. Include: - GNOME desktop with dark theme - NVIDIA GPU drivers (latest stable) - CUDA 12.x toolkit - Docker with NVIDIA container support - SSH access User: mldev with sudo and docker access Applications: Python 3.12, JupyterLab, VS Code, Git Disable screen timeout for long training runs.

Adding Tests

After OpenFactory generates the recipe, add verification tests:

Add tests to verify: - nvidia-smi returns successfully - CUDA toolkit is installed (nvcc --version) - Docker can run GPU containers (docker run --gpus all) - JupyterLab service is running

OpenFactory will add these as automated tests that run after the image builds.