This repository was archived by the owner on Aug 7, 2025. It is now read-only.
This repository was archived by the owner on Aug 7, 2025. It is now read-only.
ROCm support missing from Dockerfile #3380
Description
🚀 The feature
Currently the solution to use TorchServe with ROCm and Docker relies on separate Dockerfile Dockerfile.rocm and ROCm support is missing from Dockerfile. build_image.sh operates with Dockerfile and provides many useful build flags. Adding ROCm support for Dockerfile would make ROCm to adhere the repository standards and leverage the utility tooling created for Docker image building.
Motivation, pitch
Dockerfile.rocm relies on installing ROCm dependencies separately on top of the base image and produces large Docker images. It also doesn't have build_image.sh support. I would like to create Docker images that have ROCm support using build_image.sh script.
Alternatives
No response
Additional context
No response