Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

GPU arm64 image request #2232

Open
jerryguowei opened this issue Jul 25, 2024 · 1 comment
Open

GPU arm64 image request #2232

jerryguowei opened this issue Jul 25, 2024 · 1 comment
Labels
enhancement New feature or request

Comments

@jerryguowei
Copy link

arm64 docker image request

I couldn't find an arm64 gpu version of docker image. Is there any reason that the arm64 gpu docker image is not available? Or any people have already built the image somewhere?

AWS have arm64 gpu machine and cuda also have arm64 docker images.
@jerryguowei jerryguowei added the enhancement New feature or request label Jul 25, 2024
@frankfliu
Copy link
Contributor

We don't have aarch64 for GPU docker. The main challenge is we have to prepare native engines for PyTorch, TensorFlow,
Rust engines. We don't have contributor can work on them at the moment.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@frankfliu @jerryguowei