RTX5080's

[CAMBO84]

Won't work with RTX5080

[SuperPat45]

This error appear because RTX5000 need cuda 12.8.

The docker image need to be upgraded to the last Pytorch 2.7.0 with CUDA 12.8.

[johndpope]

i have 5090 + comfyui on older base image - make sure you're using docker + nvidia toolkit (not desktop version) - gpus /nvidia not supported with docker desktop.

UPDATE - check here-

UPDATE - sorry i mispoke - im actually using my old 3090 on this - not using 5090.
probably have to update containers.

#!/bin/bash

# Docker CE Installation Script for Ubuntu
# This script installs Docker Engine (CLI) without Docker Desktop

set -e

echo "Installing Docker CE on Ubuntu..."

# Update package index
echo "Updating package index..."
sudo apt-get update

# Install prerequisites
echo "Installing prerequisites..."
sudo apt-get install -y ca-certificates curl gnupg lsb-release

# Add Docker's official GPG key
echo "Adding Docker GPG key..."
sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg

# Add Docker repository
echo "Adding Docker repository..."
echo \
  "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu \
  $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

# Update package index again
echo "Updating package index with Docker repository..."
sudo apt-get update

# Install Docker Engine
echo "Installing Docker Engine..."
sudo apt-get install -y docker-ce docker-ce-cli containerd.io docker-compose-plugin

# Add current user to docker group
echo "Adding user to docker group..."
sudo usermod -aG docker $USER

# Start and enable Docker service
echo "Starting Docker service..."
sudo systemctl start docker
sudo systemctl enable docker

echo "Docker installation completed!"
echo "Please log out and back in (or run 'newgrp docker') to use Docker without sudo."
echo "Test with: docker --version"

#!/bin/bash

# NVIDIA Docker Runtime Installation Script
# This script installs the NVIDIA Container Toolkit for Docker GPU support

set -e

echo "🚀 Installing NVIDIA Container Toolkit for Docker..."
echo "==============================================="

# Check if running as root
if [[ $EUID -eq 0 ]]; then
   echo "❌ This script should not be run as root. Please run as a regular user with sudo privileges."
   exit 1
fi

# Check if NVIDIA GPU is available
echo "🔍 Checking for NVIDIA GPU..."
if ! nvidia-smi &> /dev/null; then
    echo "❌ NVIDIA GPU not detected or nvidia-smi not available"
    echo "Please ensure NVIDIA drivers are installed"
    exit 1
fi

echo "✅ NVIDIA GPU detected:"
nvidia-smi --query-gpu=name --format=csv,noheader

# Check if Docker is installed
echo "🔍 Checking Docker installation..."
if ! command -v docker &> /dev/null; then
    echo "❌ Docker not found. Please install Docker first."
    exit 1
fi

echo "✅ Docker found: $(docker --version)"

# Add NVIDIA GPG key
echo "🔐 Adding NVIDIA GPG key..."
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

# Add NVIDIA repository
echo "📦 Adding NVIDIA repository..."
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
  sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

# Update package lists
echo "🔄 Updating package lists..."
sudo apt-get update

# Install NVIDIA container toolkit
echo "⬇️ Installing NVIDIA Container Toolkit..."
sudo apt-get install -y nvidia-container-toolkit

# Configure Docker to use NVIDIA runtime
echo "⚙️ Configuring Docker for NVIDIA runtime..."
sudo nvidia-ctk runtime configure --runtime=docker

# Restart Docker
echo "🔄 Restarting Docker service..."
sudo systemctl restart docker

# Wait for Docker to restart
echo "⏳ Waiting for Docker to restart..."
sleep 5

# Test NVIDIA Docker runtime
echo "🧪 Testing NVIDIA Docker runtime..."
if docker run --rm --gpus all nvidia/cuda:11.0.3-base-ubuntu20.04 nvidia-smi &> /dev/null; then
    echo "✅ NVIDIA Docker runtime is working!"
else
    echo "❌ NVIDIA Docker runtime test failed"
    exit 1
fi

echo ""
echo "🎉 NVIDIA Container Toolkit installation completed successfully!"
echo "You can now run GPU-accelerated containers with Docker."
echo ""
echo "To test ComfyUI with GPU support, run:"
echo "docker compose up comfyui"
echo ""