python-itk: GPU

1

I want to use GPU based itk for image registration. I compile itk from source code, and the ITK_USE_GPU is set to ON. The compiling is OK.

The itk can be imported in python:

import itk
itk.Image[itk.F, 3]
<class ‘itkImagePython.itkImageF3’>

However, when I want to use the GPU module, bug is reported:

itk.GPUImage[itk.F, 3]
Traceback (most recent call last):
File “”, line 1, in
File “D:\Anaconda3\envs\ITK_gpu\lib\site-packages\itkLazy.py”, line 48, in getattribute
value = types.ModuleType.getattribute(self, attr)
AttributeError: module ‘itk’ has no attribute ‘GPUImage’

How can I use the GPU in python-itk?

2

Experimental GPU wrapping support is available in ITK Git master.

Experimental binary GPU Python packages are available on Linux for ITKElastix, installable with pip install itk-elastix-opencl.