Does Numpy automatically detect and use GPU?

前端 未结 4 1472
情话喂你
情话喂你 2021-02-05 03:27

I have a few basic questions about using Numpy with GPU (nvidia GTX 1080 Ti). I\'m new to GPU, and would like to make sure I\'m properly using the GPU to accelerate Numpy/Pytho

4条回答
  •  难免孤独
    2021-02-05 03:57

    JAX uses XLA to compile NumPy code to run on GPUs/ TPUs : https://github.com/google/jax

提交回复
热议问题