fix: SSH server setup and library name mismatch issues

This commit fixes two critical issues:

1. SSH Connection Failure in Docker Containers:
   - Docker backend didn't start SSH server after container creation
   - Added EnsureSSHServer() to DockerBackend similar to WSLBackend
   - Improved SSH config to explicitly bind to 0.0.0.0:22
   - Added SSHServerBackend interface and Manager.Create now calls it
   - Updated WSL backend to use consistent SSH setup script

2. Library Name Mismatch in ggo launch:
   - launch_linux.go checked for exact canonical names (libcuda.so)
   - Downloaded libraries often have different names (libaccelerator_nvidia-linux-amd64.so)
   - Now uses FindActualLibraryFiles() to scan for actual downloaded files
   - Matches pattern-based naming like studio already does

SSH server now starts automatically with:
- ListenAddress 0.0.0.0 (binds to all interfaces)
- PermitRootLogin yes
- Auto-generated host keys
- Default password (root:gpugo) for fallback auth

Library detection now handles any .so file matching vendor patterns
(libcuda*, libnvidia*, nvcuda*, nvml* for NVIDIA).

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

Author: andystime

cmd/ggo/launch/launch_linux.go

@@ -119,31 +119,26 @@ func runLaunch(args []string, shareLink, serverURL string, verbose bool) error {
 	// Get cache directory
 	cacheDir := paths.CacheDir()
 
-	// Get required libraries based on vendor
-	requiredLibs := studio.GetLibraryNames(vendor)
-
-	// Verify libraries exist after ensuring they're downloaded
-	missingLibs := []string{}
-	for _, lib := range requiredLibs {
-		libPath := filepath.Join(cacheDir, lib)
-		if _, err := os.Stat(libPath); os.IsNotExist(err) {
-			missingLibs = append(missingLibs, lib)
-		}
-	}
+	// Find actual library files that were downloaded
+	// Use FindActualLibraryFiles instead of GetLibraryNames to handle libraries
+	// with different names (e.g., libaccelerator_nvidia-linux-amd64.so)
+	actualLibs := studio.FindActualLibraryFiles(cacheDir, vendor)
+
+	if len(actualLibs) == 0 {
+		// Get canonical library names for the warning message
+		requiredLibs := studio.GetLibraryNames(vendor)
 
-	if len(missingLibs) > 0 {
 		out.Println()
 		out.Warning("Missing required libraries in cache!")
 		out.Println()
 		out.Printf("Vendor:  %s\n", vendor)
-		out.Printf("Missing: %s\n", strings.Join(missingLibs, ", "))
+		out.Printf("Missing: %s\n", strings.Join(requiredLibs, ", "))
 		out.Println()
 		out.Println("Please run:")
 		out.Println("  ggo deps sync")
 		out.Println("  ggo deps download")
 		out.Println()
-		// Continue anyway - the libraries might be named differently or not yet available
-		klog.Warningf("Missing libraries for vendor %s (continuing anyway): %v", vendor, missingLibs)
+		klog.Warningf("No libraries found for vendor %s in %s (continuing anyway)", vendor, cacheDir)
 	}
 
 	// Setup log path (consistent with ggo use)
@@ -212,9 +207,9 @@ func runLaunch(args []string, shareLink, serverURL string, verbose bool) error {
 	}
 
 	// Build LD_PRELOAD with library paths
-	if len(requiredLibs) > 0 {
+	if len(actualLibs) > 0 {
 		var preloadPaths []string
-		for _, lib := range requiredLibs {
+		for _, lib := range actualLibs {
 			preloadPaths = append(preloadPaths, filepath.Join(cacheDir, lib))
 		}
 		existingPreload := os.Getenv("LD_PRELOAD")

internal/studio/backend_docker.go

@@ -596,6 +596,58 @@ func findAvailablePort(_ int) int {
 	return SSHPortRangeMin + rand.Intn(SSHPortRangeMax-SSHPortRangeMin+1)
 }
 
+// EnsureSSHServer ensures SSH server is running in the container
+func (b *DockerBackend) EnsureSSHServer(ctx context.Context, envID string) error {
+	setupScript := `
+#!/bin/bash
+set -e
+
+# Install SSH if not present
+if ! command -v sshd &> /dev/null; then
+    if command -v apt-get &> /dev/null; then
+        apt-get update && apt-get install -y openssh-server
+    elif command -v yum &> /dev/null; then
+        yum install -y openssh-server
+    elif command -v apk &> /dev/null; then
+        apk add --no-cache openssh-server
+    fi
+fi
+
+# Configure SSH
+mkdir -p /var/run/sshd /root/.ssh
+chmod 700 /root/.ssh
+
+# Configure sshd to allow root login and listen on all interfaces
+cat > /etc/ssh/sshd_config.d/ggo-studio.conf <<'EOF'
+Port 22
+ListenAddress 0.0.0.0
+PermitRootLogin yes
+PasswordAuthentication yes
+PubkeyAuthentication yes
+UsePAM no
+EOF
+
+# Set root password if not already set (fallback for password auth)
+echo "root:gpugo" | chpasswd 2>/dev/null || true
+
+# Generate SSH host keys if they don't exist
+if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
+    ssh-keygen -A
+fi
+
+# Kill existing sshd processes (if any)
+pkill sshd || true
+
+# Start SSH server in background
+/usr/sbin/sshd
+
+echo "SSH server started"
+`
+
+	_, err := b.Exec(ctx, envID, []string{"bash", "-c", setupScript})
+	return err
+}
+
 var _ Backend = (*DockerBackend)(nil)
 
 func init() {

internal/studio/backend_wsl.go

@@ -540,11 +540,31 @@ if ! command -v sshd &> /dev/null; then
 fi
 
 # Configure SSH
-mkdir -p /var/run/sshd
-sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
-sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
+mkdir -p /var/run/sshd /root/.ssh
+chmod 700 /root/.ssh
+
+# Configure sshd to allow root login and listen on all interfaces
+cat > /etc/ssh/sshd_config.d/ggo-studio.conf <<'EOF'
+Port 22
+ListenAddress 0.0.0.0
+PermitRootLogin yes
+PasswordAuthentication yes
+PubkeyAuthentication yes
+UsePAM no
+EOF
+
+# Set root password if not already set (fallback for password auth)
+echo "root:gpugo" | chpasswd 2>/dev/null || true
+
+# Generate SSH host keys if they don't exist
+if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
+    ssh-keygen -A
+fi
+
+# Kill existing sshd processes (if any)
+pkill sshd || true
 
-# Start SSH
+# Start SSH server in background
 /usr/sbin/sshd
 
 echo "SSH server started"

internal/studio/manager.go

@@ -17,6 +17,7 @@ import (
 
 	"github.com/NexusGPU/gpu-go/internal/errors"
 	"github.com/NexusGPU/gpu-go/internal/platform"
+	"k8s.io/klog/v2"
 )
 
 var (
@@ -222,6 +223,13 @@ func (m *Manager) Create(ctx context.Context, opts *CreateOptions) (*Environment
 		return nil, err
 	}
 
+	// Setup SSH server if backend supports it
+	if sshBackend, ok := backend.(SSHServerBackend); ok {
+		if err := sshBackend.EnsureSSHServer(ctx, env.ID); err != nil {
+			klog.Warningf("Failed to setup SSH server: %v (continuing anyway)", err)
+		}
+	}
+
 	m.clearUnreachableSSH(ctx, env)
 
 	// Save environment to local state

internal/studio/types.go

@@ -239,6 +239,13 @@ type BackendSocketPath interface {
 	SocketPath(ctx context.Context) string
 }
 
+// SSHServerBackend is an optional interface for backends that support SSH server setup.
+type SSHServerBackend interface {
+	Backend
+	// EnsureSSHServer ensures SSH server is running in the environment
+	EnsureSSHServer(ctx context.Context, envID string) error
+}
+
 // SSHConfig represents an SSH configuration entry
 type SSHConfig struct {
 	Host         string