🚀 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