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

add script to isntall for GPU #122

Merged
merged 1 commit into from
Jun 12, 2024
Merged

add script to isntall for GPU #122

merged 1 commit into from
Jun 12, 2024

Conversation

qihqi
Copy link
Collaborator

@qihqi qihqi commented Jun 11, 2024

No description provided.

FanhaiLu1
FanhaiLu1 approved these changes Jun 11, 2024
View reviewed changes
Copy link
Collaborator

@FanhaiLu1 FanhaiLu1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi Han, did you test the script on GPU vms?

@wang2yn84
Copy link
Collaborator

Seems the only difference is jax[tpu] vs jax[cuda]?

@qihqi qihqi merged commit e2ee7dd into main Jun 12, 2024
4 checks passed
@qihqi qihqi deleted the hanq_fix2 branch June 12, 2024 00:11
@qihqi
Copy link
Collaborator Author

qihqi commented Jun 12, 2024

Hi Han, did you test the script on GPU vms?

yes both run_interactive and run_server ran; but perf is bad

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

@wang2yn84 wang2yn84 wang2yn84 approved these changes

@FanhaiLu1 FanhaiLu1 FanhaiLu1 approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@qihqi @wang2yn84 @FanhaiLu1