在python上玩CUDA - Numba 与 Pycuda 的比较
python上的CUDA已经广泛应用在TensorFlow,PyTorch等库中,但当我们想用GPU计算资源实现其他的算法时,不得不自己调用CUDA的python接口完成编程,以下是我在python上,利用GPU完成高斯过程计算的经验。【文首劝退】如果是想用CUDA完成较复杂的功能和算法,还是用C++实现吧。。。python的话我感觉很多已定义好的库无法正常调用,虽然numpy的很多属性和功能仍被支持,但如scikit-learn等库无法正常调用,而且坑比较多(错误提示不明显且网络上资料较少),对.
原创
2020-05-30 23:12:36 ·
5372 阅读 ·
1 评论