by Ciro Santilli (@cirosantilli, 32)
NumPy does not automatically use the GPU for it: stackoverflow.com/questions/49605231/does-numpy-automatically-detect-and-use-gpu, and PyTorch is one of the most notable compatible implementations, as it uses the same memory structure as NumPy arrays.
Sample runs on P51 to observe the GPU speedup: 
$ time ./matmul.py g 10000 1000 10000 100
real    0m22.980s
user    0m22.679s
sys     0m1.129s
$ time ./matmul.py c 10000 1000 10000 100
real    1m9.924s
user    4m16.213s
sys     0m17.293s

Ancestors

  1. PyTorch
  2. Deep learning framework
  3. Deep learning
  4. Artificial neural network
  5. Neural network
  6. Machine learning
  7. Computer
  8. Information technology
  9. Area of technology
  10. Technology
  11. Index

Discussion (0)

Sign up or sign in create discussions.

There are no discussions about this article yet.

View article source