diff options
| author | Willem Jan Palenstijn <wjp@usecode.org> | 2017-10-17 21:06:50 +0200 |
|---|---|---|
| committer | Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl> | 2017-11-01 15:53:26 +0100 |
| commit | 355ee14464f76f7a506143ef44a9b5a5319c4951 (patch) | |
| tree | 9ffafcedca289e8c7828768db0d1c9c273db3df3 /samples | |
| parent | fa92fc50355d0cbc185098ef6bcf78fa9dc65623 (diff) | |
| download | astra-355ee14464f76f7a506143ef44a9b5a5319c4951.tar.gz astra-355ee14464f76f7a506143ef44a9b5a5319c4951.tar.bz2 astra-355ee14464f76f7a506143ef44a9b5a5319c4951.tar.xz astra-355ee14464f76f7a506143ef44a9b5a5319c4951.zip | |
Move set_gpu_index, get_gpu_info from astra.astra to astra
Diffstat (limited to 'samples')
| -rw-r--r-- | samples/python/s020_3d_multiGPU.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/samples/python/s020_3d_multiGPU.py b/samples/python/s020_3d_multiGPU.py index 11a1e11..e872870 100644 --- a/samples/python/s020_3d_multiGPU.py +++ b/samples/python/s020_3d_multiGPU.py @@ -28,11 +28,11 @@ import numpy as np # Set up multi-GPU usage. # This only works for 3D GPU forward projection and back projection. -astra.astra.set_gpu_index([0,1]) +astra.set_gpu_index([0,1]) # Optionally, you can also restrict the amount of GPU memory ASTRA will use. # The line commented below sets this to 1GB. -#astra.astra.set_gpu_index([0,1], memory=1024*1024*1024) +#astra.set_gpu_index([0,1], memory=1024*1024*1024) vol_geom = astra.create_vol_geom(1024, 1024, 1024) |