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

Fix Dockerfile.spark for ARM64 architecture (DGX Spark)

Browse files

- Use NVIDIA NGC pytorch image (nvcr.io/nvidia/pytorch:24.01-py3)
- Supports ARM64/aarch64 with CUDA

🤖 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 +4 -5
docker/Dockerfile.spark CHANGED
@@ -1,11 +1,10 @@
1
- # GPU-enabled Dockerfile for DGX Spark deployment
2
- FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
3
 
4
- # Install system dependencies
5
  RUN apt-get update && apt-get install -y \
6
  curl \
7
- build-essential \
8
- git \
9
  && rm -rf /var/lib/apt/lists/*
10
 
11
  WORKDIR /app
 
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