A complete set of installation scripts to set up a GPU-accelerated AI & robotics development environment on Ubuntu 22.04. This includes NVIDIA drivers, CUDA, PyTorch, Docker, ROS 2, Miniconda, and essential development tools like Chrome, VSCode, and more.
- OS: Ubuntu 22.04 LTS (Jammy)
- GPU: NVIDIA (Tested on RTX 4090 with Driver)
- Shell: Bash
- Network: Internet connection required
| Component | Version |
|---|---|
| OS | Ubuntu 22.04 |
| GPU Driver | NVIDIA auto version |
| CUDA Toolkit | 12.2 (driver-only install) |
| PyTorch | 2.2.2 (CUDA 12.1 pip build) |
| Python | 3.10 (via Miniconda) |
| ROS2 | Humble (desktop full version) |
| Category | Script | Description |
|---|---|---|
| GPU Driver / CUDA | nvidia-driver_install.sh |
Install NVIDIA driver (autoinstall) |
cuda_install.sh |
Install CUDA Toolkit 12.2 | |
torch_install.sh |
Install PyTorch with CUDA 12.1 support | |
| Docker | Docker_install.sh |
Install Docker CE and Compose |
nvidia_docker_toolkit_install.sh |
NVIDIA Container Toolkit for GPU Docker | |
| Python Env | miniconda_install.sh |
Install Miniconda + disable base env |
| ROS2 | ROS2-humble_install.sh |
Install ROS 2 Humble (desktop full) |
| Development Tools | vscode_install.sh |
Install Visual Studio Code |
chrome_install.sh |
Install Google Chrome | |
cursor_install.sh |
Install Cursor editor | |
| Utilities | utils_install.sh |
Install Snap + Slack, Discord, Zoom, TeamViewer |
git clone https://github.com/yourname/ubuntu-auto-installation.git
cd ubuntu-auto-installationchmod +x *.sh
chmod +x scripts/*.sh# Default installation (auto-selects NVIDIA driver)
bash install_all.sh
# Or specify versions
bash install_all.sh --nvidia-driver 535 --cuda 12.2 --torch 2.2.2 --torch-cuda 12.1The install order and dependencies are automatically managed.
- NVIDIA Driver β CUDA β PyTorch
- Docker + NVIDIA Container Toolkit
- Miniconda
- ROS 2 Humble
- Chrome, VSCode, Cursor
- Utility Tools (CLI + Snap-based)
To finalize the environment:
source ~/.bashrcThen check:
nvidia-smi
conda --version
docker --version
ros2 --versionOpen two terminals:
# Terminal 1
ros2 run demo_nodes_cpp talker# Terminal 2
ros2 run demo_nodes_cpp listenerexport ROS_DOMAIN_ID=10ubuntu-auto-setup/
βββ scripts/
β βββ chrome_install.sh
β βββ cuda_install.sh
β βββ cursor_install.sh
β βββ Docker_install.sh
β βββ miniconda_install.sh
β βββ nvidia-driver_install.sh
β βββ nvidia_docker_toolkit_install.sh
β βββ ROS2-humble_install.sh
β βββ torch_install.sh
β βββ utils_install.sh
β βββ vscode_install.sh
βββ install_all.sh
βββ README.md
PRs and issues are welcome!