Spaces:
Sleeping
Sleeping
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>
- docker/Dockerfile.spark +4 -5
docker/Dockerfile.spark
CHANGED
|
@@ -1,11 +1,10 @@
|
|
| 1 |
-
# GPU-enabled Dockerfile for DGX Spark deployment
|
| 2 |
-
|
|
|
|
| 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
|