Skip to content
This repository has been archived by the owner on Jun 7, 2023. It is now read-only.

GPU support #46

Open
2 of 3 tasks
hans opened this issue May 25, 2020 · 1 comment
Open
2 of 3 tasks

GPU support #46

hans opened this issue May 25, 2020 · 1 comment
Labels
cli enhancement New feature or request models Platform-general model wrapper issues

Comments

@hans
Copy link
Collaborator

hans commented May 25, 2020 •
edited
Loading

Use some new environment variable e.g. LMZOO_GPU_ENABLED.

  • Verify that CUDA_VISIBLE_DEVICES gets passed through on all backends
  • update images to read and act on env variable
  • add CLI switch
@hans hans added enhancement New feature or request cli models Platform-general model wrapper issues labels May 25, 2020
@hans hans mentioned this issue May 26, 2020
Closed
@AnneBeyer
Copy link

Could you please add some more instructions on how to enable GPU usage for the other models not mentioned in #49?
We're trying to use some larger pre-trained models with syntaxgym and can't get them to use the available GPUs.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
cli enhancement New feature or request models Platform-general model wrapper issues
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@hans @AnneBeyer