Configure GPU index for 'astra_cuda', select GPU currently used by PyTorch in OperatorModule #1546

Open

jleuschn wants to merge 4 commits into

odlgroup:masterfrom

jleuschn:master

Commits on Mar 24, 2020

introduce gpu_index for astra_cuda backend and RayTransform
jleuschn
committed

Commits on Mar 25, 2020

run RayTransform on torch.cuda.current_device() if wrapped by OperatorModule
jleuschn
committed
fix: add check if cuda is available in OperatorModule
jleuschn
committed

Commits on Nov 4, 2020

fix _resize_discr in case original discr has nodes_on_bdry=True on non-affected axis
jleuschn
committed