PyCUDA lets you access Nvidia‘s CUDA parallel computation API from Python
conda install powerai::pycuda