Use Metal Performance Shaders for hardware acceleration if available. #12

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

Open

amarkoolk wants to merge 1 commit into linklab-uva:main from amarkoolk:training/use-mps

amarkoolk commented Jan 28, 2025

Allow torch to use mps if cuda is not available.

Works for training and evaluation.


          allow model to use mps if available

b0d1232

Collaborator

john-chrosniak commented Feb 12, 2025

@amarkoolk Could you also add the mps changes to mpc/run_nmpc_orca_deep_pacejka.py and mpc/run_nmpc_orca_deep_dynamics.py?

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

Labels

None yet