Computación científica con Python con GPU modernas con doble precisión

14

¿Alguien ha usado computación científica de doble precisión con GPU de nueva generación (por ejemplo, K20) a través de Python?

Sé que esta tecnología está evolucionando rápidamente, pero ¿cuál es la mejor manera de hacer esto actualmente? GPU está fuera del alcance de las bibliotecas científicas populares de Python numpy y scipy, y quería usar theano pero parece que solo usa precisión float32 para GPU. Soy consciente de que Google puede proporcionar resultados de búsqueda para Python GPU, pero espero obtener más información que una lista de proyectos que pueden o no estar en camino de conocer a su creador.

clíper
fuente
44
Si lo necesitara, probablemente usaría PyOpenCL . La codificación de GPU de uso general todavía es de nivel bastante bajo (prueba la interfaz OpenCL C, es difícil). Sin embargo, PyOpenCL parece abstraer tanto como sea posible y parece tener un impulso considerable detrás de él.
boyfarrell
1
"nueva generación" debería ser bastante similar a la serie anterior de Tesla con doble precisión, y probablemente la única advertencia sea la versión CUDA / driver. Por lo tanto, los métodos de doble precisión que funcionan con Tesla (por ejemplo, M2070) y la versión actual de CUDA / Driver también deberían funcionar con el K20.
internetscooter
1
Tal vez aquí ( stackoverflow.com/questions/5957554/python-gpu-programming ) pueda brindarle ayuda.
eusoubrasileiro
2
Theano tiene un nuevo back-end de GPU que admite float64. Todavía no está completo, pero lo anunciaremos en estado beta esta semana.
nouiz
Gracias @nouiz: te sugiero agregar tu comentario como respuesta cuando realices el lanzamiento.
Aron Ahmadia

Respuestas:

3

No sé por qué pongo esta respuesta en un comentario ...

Si lo necesitara, probablemente usaría PyOpenCL . La codificación de GPU de uso general todavía es de nivel bastante bajo (prueba la interfaz OpenCL C, es difícil). Sin embargo, PyOpenCL parece abstraer tanto como sea posible y parece tener un impulso considerable detrás de él.

boyfarrell
fuente
Eh, sucede. A veces hago lo mismo.
Geoff Oxberry