gary-boon Claude Opus 4.5 commited on
Commit
a4927aa
·
1 Parent(s): 9d00d33

Fix Dockerfile.spark for CUDA 13.0 compatibility

Browse files

- Use nvidia/cuda:12.4.0-runtime-ubuntu22.04 base image
- Install PyTorch with CUDA 12.4 wheels (compatible with CUDA 13.0 driver)
- Build Python dependencies properly for ARM64

🤖 Generated with [Claude Code](https://claude.com/claude-code)

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

Files changed (1) hide show
  1. docker/Dockerfile.spark +17 -7
docker/Dockerfile.spark CHANGED
@@ -1,17 +1,27 @@
1
- # GPU-enabled Dockerfile for DGX Spark deployment (ARM64/aarch64)
2
- # Using NVIDIA NGC PyTorch image which supports ARM64
3
- FROM nvcr.io/nvidia/pytorch:24.01-py3
4
 
5
- # Install additional system dependencies
6
  RUN apt-get update && apt-get install -y \
 
 
 
7
  curl \
8
- && rm -rf /var/lib/apt/lists/*
 
 
9
 
10
  WORKDIR /app
11
 
12
- # Copy and install Python requirements
 
 
 
 
 
 
13
  COPY requirements.txt .
14
- RUN pip install --no-cache-dir -r requirements.txt
15
 
16
  # Copy backend code
17
  COPY backend/ ./backend/
 
1
+ # GPU-enabled Dockerfile for DGX Spark deployment (ARM64/aarch64 + CUDA 13.0)
2
+ FROM nvidia/cuda:12.4.0-runtime-ubuntu22.04
 
3
 
4
+ # Install Python and system dependencies
5
  RUN apt-get update && apt-get install -y \
6
+ python3 \
7
+ python3-pip \
8
+ python3-dev \
9
  curl \
10
+ build-essential \
11
+ && rm -rf /var/lib/apt/lists/* \
12
+ && ln -s /usr/bin/python3 /usr/bin/python
13
 
14
  WORKDIR /app
15
 
16
+ # Install PyTorch with CUDA support first (before other requirements)
17
+ # Using CUDA 12.4 wheels which are compatible with CUDA 13.0 driver
18
+ RUN pip3 install --no-cache-dir \
19
+ torch \
20
+ --index-url https://download.pytorch.org/whl/cu124
21
+
22
+ # Copy and install remaining Python requirements
23
  COPY requirements.txt .
24
+ RUN pip3 install --no-cache-dir -r requirements.txt
25
 
26
  # Copy backend code
27
  COPY backend/ ./backend/