Rodando script Python na GPU

Este script de exemplo eu fiz para demonstrar de forma rápida a diferença entre executar um script com alta complexidade na CPU x GPU.
Baixando o código fonte
jaccon/python-running-gpu
The example to running the Python script in GPU. Contribute to jaccon/python-running-gpu development by creating an account on GitHub.
Comparando tempos em CPU e GPU
Neste exemplo estamos comparando o tempo de execução de uma operação matemática complexa em tempo de CPU e também na GPU.
Neste exemplo do screenshot abaixo, estou rodando o script em uma CPU Core i5 9º geração e uma GPU RX 580 de 8GB.

No meu teste aqui o tempos de execução foram bem diferentes:
Rodando na CPU: 21.32215951703256
Rodando na GPU: 0.20966976397903636