Hi, are there any plans to construct a Docker image with both TF and Torch installed? Currently, the images are built for either TF or Torch but not both together. I am pulling the HF Docker images into Singularity containers but am needing to install additional packages using the Singularity recipe & it will save time if both packages are incorporated into the single image.
Related topics
Topic | Replies | Views | Activity | |
---|---|---|---|---|
Containerizing Huggingface Transformers for GPU inference with Docker and FastAPI | 0 | 2941 | October 5, 2021 | |
How to deal with no GPU during docker build time | 4 | 10135 | October 23, 2023 | |
Need Example Docker File with GPU Support | 4 | 2025 | October 12, 2023 | |
DiffusionPipeline in docker but GPU from user machine | 0 | 142 | June 8, 2024 | |
New Spaces Primitive Feature Request - Gradio plus Torch plus Transformers Docker spin up? | 1 | 357 | March 25, 2023 |