从spyder在GPU上实现python代码

时间:2018-06-04 09:09:51

标签: python cuda

根据知识with tf.device('/GPU')可用于在GPU中实现张量流。是否有任何类似的方法在GPU(Cuda)上实现任何python代码?或者我应该使用pycuda?

1 个答案:

答案 0 :(得分:1)

对于python中的并行处理,需要在代码和用于并行执行的gpu / cpu之间的某些中间库或包。一些流行的软件包是 pycuda,numba 等。如果你想使用简单的python语法进行gpu编程而不使用其他框架,比如tensorflow,那么看看this