Open gfardell opened 2 years ago
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.
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.