• Joseph Young's avatar
    Refactor Dockerfile for CPU and CUDA builds · c004ecdc
    Joseph Young authored
    Switched to Chainguard images as base for both CPU and CUDA backend builds for improved security and compatibility. Replaced Ubuntu base with Chainguard's Python image for the CPU builds and PyTorch CUDA image for GPU acceleration, resolving python requirements conflicts. Updated package installation commands to align with the new Redhat-compatible base images. The Dockerfile now installs only the necessary dependencies, as Python is provided by the base image.
    
    These changes will facilitate a more secure and streamlined build process with better dependency management across different platforms.
    c004ecdc
Dockerfile-cuda 3.78 KB