Skip to content

fixed dockerfile for arm64 systems#44603

Open
michalrzak wants to merge 1 commit intohuggingface:mainfrom
michalrzak:docker-fix
Open

fixed dockerfile for arm64 systems#44603
michalrzak wants to merge 1 commit intohuggingface:mainfrom
michalrzak:docker-fix

Conversation

@michalrzak
Copy link
Copy Markdown

What does this PR do?

A quick fix that updates the Dockerfile to run on arm64 systems (such as the NVIDIA Spark). The previous version of the Dockerfile fails on arm64 systems due to SudachiPy, which only provides wheels for x86_64 architectures and must be built from source, which requires the Rust compiler. This PR adds the Rust compiler to the Docker container if the architecture is arm64.

Further, torchcodec==0.8.0 is not available on arm64. To circumvent this, the latest torchcodec version is installed if the system is specified as arm64.

@ydshieh

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant