This repository has been archived by the owner on Aug 1, 2022. It is now read-only.

ASTRA multi GPU usuage #105

Open

gfardell opened this issue Feb 2, 2022 · 0 comments

Member

gfardell commented Feb 2, 2022 •

edited

Loading

As default ASTRA seems to only use one GPU. Multiple can be set using:

astra.astra.set_gpu_index([0,1])

But it may be worth adding in some logic to our wrappers to query the available devices and make use of them.

The text was updated successfully, but these errors were encountered:

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