diff options
| -rw-r--r-- | python/astra/astra_c.pyx | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/python/astra/astra_c.pyx b/python/astra/astra_c.pyx index 1d3fad7..65dbf28 100644 --- a/python/astra/astra_c.pyx +++ b/python/astra/astra_c.pyx @@ -84,6 +84,8 @@ IF HAVE_CUDA==True: if use_cuda()==True: if not isinstance(idx, collections.Iterable) or isinstance(idx, six.string_types + (six.text_type,six.binary_type)): idx = (idx,) + if memory < 1024*1024: + raise ValueError("Setting GPU memory lower than 1MB is not supported.") params.memory = memory params.GPUIndices = idx setGlobalGPUParams(params) |