100x Faster Than NumPy... (GPU Acceleration)

Ғылым және технология

CHECK OUT MY NEW UDEMY COURSE, NOW 90% OFF WITH THIS CODE:
www.udemy.com/course/python-s...
Code:
github.com/lukepolson/youtube...
Old Video:
• Billiard Balls and the...
Get Your Billy T-Shirt: my-store-d2b84c.creator-sprin...
Discord: / discord
Instagram: / mrpsolver

Пікірлер: 101

  • @MrPSolver
    @MrPSolver Жыл бұрын

    UPDATE: Thanks to @swni on Reddit for the suggestion to use the `ids_pairs` array to index to get `x_pairs` and `y_pairs` as opposed to reusing the `torch.combinations` function. This reduces the simulation time required for 10000 particles to only 20 seconds (about half what is shown in the video). Code has been updated on GitHub! To compare NumPy and PyTorch fairly under these new conditions, I simulate 5000 particles in each case. PyTorch takes 6.3 seconds to run (remember, it also has around a 2 second overhead), while NumPy takes about 823 seconds, indicative of about a 100x increase.

  • @ibonitog

    @ibonitog

    Жыл бұрын

    Could you test CuPy please?

  • @MrHaggyy

    @MrHaggyy

    Жыл бұрын

    There must still be a lot of potential. A GPU calculates 1080x1920 ~2Mio RGB value per frame. You don`t need to check n^2 combinations for collision, n! should be enough because if P1 collides with P2, P2 also collides with P1. Especially something like checking for collision can be blazingly fast on a GPU. Your 3070 has over 5000 cores and each one has SIMD instructions. So you can do about 20k fp ops per clk. I would check the particles for collision when creating the pairs. You have the function anyway so it`s an easy-to-fix bug.

  • @fabiopimentel6981
    @fabiopimentel6981 Жыл бұрын

    Bro please never stop doing physics videos, they are amazing! I know they are not the most popular videos in your channel but they are super helpful for someone that only had one programming subject and was with Fortran :( . Greetings from the Dominican Republic! haha

  • @ConstantlyDamaged
    @ConstantlyDamaged Жыл бұрын

    It always catches me off guard to see non-meme videos from you. I am more into web-interacting services rather than data manipulation/science-so async is my wheelhouse rather than this stuff. Still fascinating to watch.

  • @zb9458
    @zb9458 Жыл бұрын

    Awesome vid! Love seeing Pytorch being leveraged for its first class GPU support for things other than machine learning. If I recall correctly, someone had a blog post about using pytorch to optimize a shape for rolling (i.e. reinventing the wheel) and it used pytorch, super funny, but cool. Great video!!

  • @tiberioborgesvale3213
    @tiberioborgesvale3213 Жыл бұрын

    Congrats! Great video! Please, don't stop, your videos are incredibly didactic! I allways cite your channel to my students in my Classical Dynamics classes.

  • @gabrielzinatorosa3024
    @gabrielzinatorosa3024 Жыл бұрын

    Very interesting content and I really appreciate the way you show both notebooks side by side to compare the results. Thank you very much!

  • @Louis-ml1zr
    @Louis-ml1zr Жыл бұрын

    Nice I've been waiting for this one ! thanks , looking forward to seeing the next ones

  • @samlaki4051
    @samlaki4051 Жыл бұрын

    first non-humour vid i see and its awesome! will try to learn more! thank you professor!

  • @Nooreazy
    @Nooreazy Жыл бұрын

    I'm honored that your stuff comes up on my feed. Amazing work!

  • @EVL624
    @EVL624 Жыл бұрын

    I am a numerical physicist, and this will be very helpful for me. I am currently running all my simulations om CPU (though using MPI for parallellization)

  • @geoffreyanderson4719

    @geoffreyanderson4719

    Жыл бұрын

    Please learn about nvblas and openblas and code vectorization in my other comment here today. The two keys are writing vectorized numpy or pandas code, plus activating the nvblas or openblas subsystem. Let me know if want help.

  • @jawad9757

    @jawad9757

    Жыл бұрын

    You may want to take a look at CUDA C++ if you have Nvidia GPU(s) and are concerned with performance

  • @Impo666
    @Impo666 Жыл бұрын

    I absolutely love this. I'm making my own game engine (fun hobby, tbh) with OpenGL, numpy and Python, and for some time I've thought about where to simulate my physics. This is an eye-opener, and it looks fun as heck! Espec. the matplotlib animation for some "lazy" collision simulations. This vid brings me straight to my college days

  • @Ilya-iu5ih
    @Ilya-iu5ih Жыл бұрын

    Great video, thanks! Consider using indexing by the coordinates of particles in space. The idea is that the coordinates of the particles are rounded to the size of the box, and the collision check occurs only for those particles that are inside the same box. This usually reduces the number of pairs by 90%.

  • @antonioarroyopolonio2520
    @antonioarroyopolonio2520 Жыл бұрын

    Amazing content. I had a professor when I was in the physics degree that told us about the power of GPU when coding "big numbers". The GPUs have up to 1000 more ("dumb") cores than the CPU and that can be really powerfull. I am now working on my PhD and I use python to do the work. I think that I can learn a lot from you. Thank you!

  • @MyMinecrafter97
    @MyMinecrafter97 Жыл бұрын

    Very interesting, would love to see more of this!

  • @bwallace1598
    @bwallace1598 Жыл бұрын

    Great educational video, mate! I'm a CS Grad student and was beginning to get to the later ML courses. Your explanation and side-by-side logic demonstration with Numpy convinced me to do a bit of research and switch from TF to Pytorch! Thanks so much!! I eagerly look forward to the next video!

  • @Thierry080
    @Thierry080 Жыл бұрын

    Super cool, your meme videos are hilarious but this quality content is why I subbed in the first place

  • @shamaldesilva9533
    @shamaldesilva9533 Жыл бұрын

    Dude a GPU accelarated python series would be amazing 😍😍😍

  • @vialomur__vialomur5682
    @vialomur__vialomur5682 Жыл бұрын

    Wow this is really interesting! Thanks! Waiting for more videos

  • @tiddlywinks497
    @tiddlywinks497 Жыл бұрын

    Oh damn, this is what my thesis is on! Good to see that some great resources are being put out for it

  • @alexdefoc6919
    @alexdefoc691910 ай бұрын

    I hope you continued this series

  • @calebwhales
    @calebwhales Жыл бұрын

    Great to see ya again mate

  • @tuomassavolainen218
    @tuomassavolainen218 Жыл бұрын

    Nice video, did not think about using pytorch to replace Numpy, but it makes perfect sense for parellelizing numpy code👍. Just a quick tip for additional speedup. Instead of comparing the distance directly you can compare the squared distance for collision detection, this avoids using the square root function which is "slow" at least compared to all the dot products, though it might not matter much for simulations of this scale.

  • @tunafllsh
    @tunafllsh Жыл бұрын

    GPU takes advantage of linear operations. So I'm not really sure, but if you use some data structures like quadtree the complexity of the computation might drastically simplify. And you won't need to calculate all n² distances. In fact most particles are not collading with each other. One need to test it, but with that CPU might still outperform the overhead of the GPU, since there won't be that many computations.

  • @shinrinyuku4972
    @shinrinyuku4972 Жыл бұрын

    Very good intro video into GPU programming, it even gave me couple of ideas. One question if I may. Why wouldn't you do the simulation with event driven algorithm, since that would save a lot of resources and you can avoid overlaps of particles (ie the need to choose small timesteps). I get this is a tutorial/introduction video, but that implementation would be very interesting as well!

  • @monkeyofwar6818
    @monkeyofwar6818 Жыл бұрын

    It's so relaxing so see someone else's explanation. I'm so tired of doing work in graduate school XD

  • @Andres186000
    @Andres186000 Жыл бұрын

    Thank you so much , I was already using PyTorch for something, but I couldn't figure out how to create the equivalent of the "x_pairs" array I needed to use, thanks.

  • @govindtuli2913
    @govindtuli2913 Жыл бұрын

    Multiprocessing library also helps to utilize all available threads. I was generating a mandlebulb and it went from 4 minutes to 1 minute when I optimized code for using it.

  • @LydellAaron
    @LydellAaron Жыл бұрын

    Sweet topic. Thank you!

  • @tclin77
    @tclin77 Жыл бұрын

    Definitely a must watch!

  • @geoffreyanderson4719
    @geoffreyanderson4719 Жыл бұрын

    Nvblas can be used by numpy. Nv stands for nvidia. Just configure your host a bit, which is easy. Openblas can also be used by numpy, which is more common. By default, your linux is using a gnu blas which is super slow by comparison. Nvblas uses the gpu for the linear algebra operation s in your numpy code. Just be sure to write vectorized numpy code , not for loops. You don't change your application code at all, which is a big benefit for ease of maintenance. Openblas will recruit all your cpu cores and implicitly parallelize your matrix matg, greatly speeding it up, as will nvblas.

  • @seelooooo
    @seelooooo Жыл бұрын

    Nice video! I’ve also got a question in the part of calculating whether particles collide with each other. Is there any advantage of the video’s method compared to use: DIS = torch.cdist(points,points) DIS = torch.triu(DIS, diagonal = 1) Pairs = DIS.nonzero() Or, they are having the same computational complexity?

  • @MrPSolver

    @MrPSolver

    Жыл бұрын

    Never seen "torch.cdist" before! Thank you for this comment. Huge reason why I post videos like this...to learn more from the comments :)

  • @s.v.8662
    @s.v.8662 Жыл бұрын

    Could another distribution really come up for different potentials around the particles? I thought of the Boltzmann distribution as a thermodynamic necessity due to maximization of entropy

  • @btr_Z
    @btr_Z Жыл бұрын

    Thanks, now I finally will have one reason to tell my dad to buy me a graphics card😂

  • @PaawanS
    @PaawanS Жыл бұрын

    This is brilliant!

  • @fizik_amorim
    @fizik_amorim Жыл бұрын

    Thanks for the video ! You are awesome ! I have a question. Is it possible to use pytorch to optimize a code with a lot of functions from scipy ? Like solving a lot of differential equations, nonlinear equations, interpolating and integrating functions all in one big code. I'm currently optimizing my code with the joblib library to run it in parallel.

  • @Ilya-iu5ih
    @Ilya-iu5ih Жыл бұрын

    torch by default includes in each tensor the telemetry necessary to calculate the derivative for the error back propagation algorithm. Use the require_grad=False parameter, this will speed up the calculation even more. x = torch.randn(3, requires_grad=False)

  • @filippavlove5574
    @filippavlove5574 Жыл бұрын

    love it, keep it up :)

  • @obliquesealray2188
    @obliquesealray2188 Жыл бұрын

    thank you for this!

  • @daze8410
    @daze8410 Жыл бұрын

    Just recently got familiar with multithreading so I guess this is the natural progression

  • @LS-xb2fh
    @LS-xb2fh Жыл бұрын

    If this is all about optimization, you should probably compare the sqared distance between particles to eliminate the need to calculate the square root. :)

  • @cw9249
    @cw9249 Жыл бұрын

    perfect intro to torch for someone who is familiar with numpy

  • @gdmitrich
    @gdmitrich Жыл бұрын

    thats so cool need this more in field of quantum chemistry❤❤

  • @sergiolucas38
    @sergiolucas38 Жыл бұрын

    Great video :)

  • @Omar-jn9zf
    @Omar-jn9zf Жыл бұрын

    12:00 why don't you simply use torch.cdist (if you have a batch of vectors, otherwise use torch.pdist) which calculates the p-norm (p=2 in your case) distance between each pair of the two collections of row vectors. This is supposed to be much faster than your code, even though I didn't test it.

  • @arduous222
    @arduous222 Жыл бұрын

    I frequently do so-called "model-fitting" using MCMC (or anything good enough), where each set of data consists of 1k-10k data. I wonder whether this could benefit from GPU acceleration or the overhead would be too much.

  • @bean_mhm
    @bean_mhm Жыл бұрын

    this is really cool!

  • @galenseilis5971
    @galenseilis59717 ай бұрын

    Why was it 'bad' that some of particles were colliding in the initial conditions?

  • @m_sedziwoj
    @m_sedziwoj Жыл бұрын

    Maybe I add that you can use AMD GPUs but currently only in Linux (as Nvidia have CUDA, AMD have ROCm)

  • @user-om4by2ig8g
    @user-om4by2ig8g Жыл бұрын

    can you suggest me books for this relevant problems of laplace transform via python

  • @habtiephys
    @habtiephys Жыл бұрын

    I think operations on Pythorch Tensor are also faster than on Numpy arrays both on cpu

  • @nieczerwony
    @nieczerwony Жыл бұрын

    It's all great but one thing. I don't need your face (person) sitting in fron do 2 screens and covering them 😂

  • @TZ-nd1cm
    @TZ-nd1cm Жыл бұрын

    What are libraries that must be imported?

  • @apoorvvyas52
    @apoorvvyas52 Жыл бұрын

    Great video

  • @sohamtilekar5126
    @sohamtilekar51263 ай бұрын

    Can You Make video on the PyOpenCl

  • @jorgesaxon3781
    @jorgesaxon3781 Жыл бұрын

    PLEASE MAKE A TUTORIAL ON HOW TO HANDLE BIG INTEGERS (>64INT) ON THE GPU 🙏

  • @void2258
    @void2258 Жыл бұрын

    What program are you using here that let's you put notes in the code like this?

  • @MrPSolver

    @MrPSolver

    Жыл бұрын

    VSCode and Jupyer Notebook!

  • @kedaracholkar6055
    @kedaracholkar6055 Жыл бұрын

    GTX 3070? Do you mean rtx?

  • @MrPSolver

    @MrPSolver

    Жыл бұрын

    Haha ya 😂

  • @maelstrom254
    @maelstrom254 Жыл бұрын

    Torch jit and torch compile is a lot faster than just torch

  • @baldpolnareff7224
    @baldpolnareff7224 Жыл бұрын

    So my intuition of rewriting stuff in pytorch just for fun was not unreasonable after all!

  • @harikrishnanb7273
    @harikrishnanb7273 Жыл бұрын

    2 minutes into the video, what about the performance of pytorch compared to numpy in CPU? is it faster there also !? have you tried numba !?

  • @alexdefoc6919
    @alexdefoc691910 ай бұрын

    Oh so good

  • @frankkoslowski6917
    @frankkoslowski6917 Жыл бұрын

    Nice sales pitch for Microsoft Visual Studio. Had `cuda` up and running nicely in a previous installation of Visual Studio. All of that had been with C++ in mind. So Python was not really considered at that time. Was hoping to do the same with PyCharm and Browser-based Notebook using Python exclusively. That's when it got confusing to the point of dropping the idea. 😒

  • @ibonitog
    @ibonitog Жыл бұрын

    What about cupy (CUDA drop-in replacement for numpy)? Is the performance uplift comparable to pytorch?

  • @teslapower220

    @teslapower220

    Жыл бұрын

    Yes...

  • @arunsebastian4035
    @arunsebastian4035 Жыл бұрын

    Very nice video

  • @nikolasfrancorios312
    @nikolasfrancorios312 Жыл бұрын

    "GPU, wich most people have acess today" Looks like we've got some serious worldknowing issue going on here

  • @MrPSolver

    @MrPSolver

    Жыл бұрын

    Lots of free nodes you can use here that have access to GPU resources: colab.research.google.com/

  • @pauldirac6243
    @pauldirac62438 ай бұрын

    When I try to run your code, I get the error message: No module named 'torch' What am I doing wrong?

  • @holthuizenoemoet591
    @holthuizenoemoet591 Жыл бұрын

    Would this work on a RX 6800 or intel Ark 770?

  • @baldpolnareff7224

    @baldpolnareff7224

    Жыл бұрын

    If I remember correctly pytorch runs on AMD and intel arc as well

  • @mikejohnston9113
    @mikejohnston9113 Жыл бұрын

    I can't get accepted into your discord. In the two lines ani.save()... I get a file not found exception. I am using Python 3.11.3. I really like the article and video. Thanks

  • @mikejohnston9113

    @mikejohnston9113

    Жыл бұрын

    I found the problem, I hand not installed python-ffmpeg. It is fixed now. Thanks

  • @fnegnilr10
    @fnegnilr10 Жыл бұрын

    What!!!!!!!!!!!!!!!!!!!!!!!!!

  • @infiniteflow7092
    @infiniteflow7092 Жыл бұрын

    Does Pytorch have numerical integration capabilities?

  • @user-xh9pu2wj6b

    @user-xh9pu2wj6b

    Жыл бұрын

    there's torch.trapz which does exactly that

  • @infiniteflow7092

    @infiniteflow7092

    Жыл бұрын

    @@user-xh9pu2wj6b That's cool. Any idea how its accuracy compares to say scipy.quad function?

  • @mathiaz943
    @mathiaz943 Жыл бұрын

    Does GPU mean NVDIA GPU specifically? Will we ever have libraries utilizing ANY general GPU?

  • @bryce07

    @bryce07

    Жыл бұрын

    it works with any GPU

  • @c0d3w4rri0r
    @c0d3w4rri0r Жыл бұрын

    I mean you start by saying most people have access to a GPU these days and this is absolutely true. But plenty don't have an NVIDIA GPU and as I understand it pytorch doesn't support non NVIDIA gpus? might be worth re writing this with pyopencl.

  • @antonioparesini912
    @antonioparesini912 Жыл бұрын

    Where billy

  • @tsunningwah3471
    @tsunningwah3471 Жыл бұрын

    wowowowoowwww

  • @TheBuilder
    @TheBuilder Жыл бұрын

    0:03 "If you coded in python before" while showing a screen full of braces

  • @Zucih1
    @Zucih1 Жыл бұрын

    Hey Nvidia did i miss the new gtx 3070 !!

  • @Singlton
    @Singlton Жыл бұрын

    Gtx 3070? Is this from china!? 🤣😂

  • @haiminh256
    @haiminh256 Жыл бұрын

    i'll break your comment bar with C++

  • @ssmith5048
    @ssmith5048 Жыл бұрын

    soo instead of using the poor man´s version of Fortran for such calculations, just use Fortran. It is not only perfect for arrays but also natively parallel. You can even make a python wrapper if you want some gui to please the eye. But , I get it, it would not be cool for the kids on youtube...but if you really need efficiency , give it a try.

  • @hoaanduong3869
    @hoaanduong3869 Жыл бұрын

    Damn, he used a Deep Learning Framework to replace Numpy, a Mathematics Framework :v

  • @Omar-jn9zf
    @Omar-jn9zf Жыл бұрын

    The entire PART 1 can be more efficiently rewritten in one line: d_pairs = torch.pdist( r.T )

  • @haiminh256
    @haiminh256 Жыл бұрын

    #Include int main() { std::cout

  • @inxendere
    @inxendere Жыл бұрын

    the RTX 3070 is already considered mid range??!! 🥲

  • @perlindholm4129
    @perlindholm4129 Жыл бұрын

    Use transfer function its billion times faster: h = np.random.rand(10000) idx = np.arange(10000) X = X_train[idx].astype("float32")/255.0 yt = y_train[idx] + 1 # 1..10 x = X.mean(1) ids = np.argsort(x) i=0 while True: err = yt[ids] - x[ids] * h h += 0.1*err print(np.mean(err**2)) i+=1 if np.mean(err**2)

Келесі