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.

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

Enable GPU support #62

Closed
akapet00 opened this issue Aug 15, 2021 · 0 comments
Closed

Enable GPU support #62

akapet00 opened this issue Aug 15, 2021 · 0 comments

Comments

@akapet00
Copy link
Member

Enable the use of GPU for training neural density estimator.

The GPU is expected be effective only for large neural networks with operations that are inherently fast on the GPU, e.g., convolutions. No speed ups in training time will be achieved for the default architectures provided in sbi.

@akapet00 akapet00 self-assigned this Aug 15, 2021
@akapet00 akapet00 added this to To do in sbi integration (GSoC 2021) via automation Aug 15, 2021
sbi integration (GSoC 2021) automation moved this from To do to Done Aug 15, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Development

No branches or pull requests

1 participant