Skip to content

Improve Docker image of rocm flashinfer published at https://hub.docker.com/r/rocm/flashinfer #80

Description

@diptorupd

The current docker image has two design issues:

a) The sources are part of the image and can potentially lead to issues if a user of the image changes directory and gets in the source directory. The sources should be removed from the published image as it is primarily meant for end users.

b) The amd-flashinfer wheel package is installed in a custom micromamba env requiring end users to first switch to the right env. Ideally, the package should be installed in the base image.

Metadata

Metadata

Assignees

No one assigned

    Labels

    dockerUpdate CI or developer docker imageinfrastructureInfrastructure upgrade

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions