Common CUDA base image for AI inference services with multi-architecture support
2.4K
A multi-architecture Docker base image optimized for AI inference services, providing common system dependencies and Python packages for CUDA-accelerated applications. Supports both Intel/AMD x64 systems and ARM64 NVIDIA Jetson devices.
# Pull the latest image
docker pull phygrid/cuda-base:latest
# Use as base image in your Dockerfile
FROM phygrid/cuda-base:1.0.0
build-essential, cmake, git, wget, curllibasound2-dev, portaudio19-dev, libsndfile1, ffmpeglibgl1, libglib2.0-0, OpenCV dependenciespatchelf for executable stack fixesca-certificates and utilitiesaiofiles, python-dotenv, python-multipartappuser with proper permissions/app/{cache,models,data,logs} directoriesRepository: phygrid/cuda-baseโ
latest - Latest stable release1.0.0, 1.0.1, etc. - Specific semantic versionslinux/amd64, linux/arm64FROM phygrid/cuda-base:1.0.0
# Copy your application
COPY . /app/
# Install additional dependencies
RUN pip install -r requirements.txt
# Override default command
CMD ["python", "main.py"]
# Run interactive container
docker run -it --rm \
-v $(pwd):/app/workspace \
-p 8000:8000 \
phygrid/cuda-base:latest \
bash
# Run with GPU support (Intel/AMD systems)
docker run -d \
--name my-ai-service \
--gpus all \
-p 8000:8000 \
-v /data:/app/data \
phygrid/cuda-base:latest
# Run on NVIDIA Jetson devices
docker run -d \
--name my-ai-service \
--runtime nvidia \
--gpus all \
-p 8000:8000 \
-v /data:/app/data \
phygrid/cuda-base:latest
# Clone repository
git clone https://github.com/phygrid/cuda-base.git
cd cuda-base
# Build image
docker build -t phygrid/cuda-base:custom .
# Build for multiple architectures
docker buildx build --platform linux/amd64,linux/arm64 -t phygrid/cuda-base:custom .
This project uses automated semantic versioning:
VERSION file for major/minor bumpsv1.0.0)See DOCKER_DEPLOYMENT.mdโ for detailed deployment workflow.
The image includes a built-in health check:
# Test health endpoint
docker run --rm phygrid/cuda-base:latest python /app/health_check.py
We welcome contributions! Please:
git checkout -b feature/amazing-feature)git commit -m 'Add amazing feature')git push origin feature/amazing-feature)# Clone and setup
git clone https://github.com/phygrid/cuda-base.git
cd cuda-base
# Test build locally
docker build -t phygrid/cuda-base:test .
docker run --rm phygrid/cuda-base:test python /app/health_check.py
This project is licensed under the MIT License - see the LICENSEโ file for details.
| Variable | Default | Description |
|---|---|---|
PYTHONUNBUFFERED | 1 | Disable Python output buffering |
PYTHONDONTWRITEBYTECODE | 1 | Don't write .pyc files |
PIP_NO_CACHE_DIR | 1 | Disable pip cache |
PIP_DISABLE_PIP_VERSION_CHECK | 1 | Skip pip version checks |
The image includes standard OCI labels:
LABEL org.opencontainers.image.title="Phygrid CUDA Base"
LABEL org.opencontainers.image.description="Common CUDA base image for AI inference services"
LABEL org.opencontainers.image.vendor="Phygrid"
LABEL org.opencontainers.image.version="1.0.0"
Content type
Image
Digest
sha256:96aa88d0bโฆ
Size
17.3 GB
Last updated
about 1 year ago
docker pull phygrid/cuda-base