AI for all: Real skills that transform every role. Our AI Academy is here.

AI for all: Real skills that transform every role. Our AI Academy for enterprises and individuals is here.

    Get More Info
    Jeremy Howard Photo

    Jeremy Howard

    FAST.AI / PLATFORM.AI

    San Francisco

    Talk: Using GPU acceleration with Pytorch to make your algorithms 2000% faster

    Most developers are aware that some algorithms can be run on a GPU, instead of a CPU, and see orders of magnitude speedups. However, many people assume that:

    • Only specialist areas like deep learning are suitable for GPU

    • Learning to program a GPU takes years of developing specialist knowledge

    It turns out that neither assumption is true! Nearly any non-recursive algorithm that operates on datasets of 1000+ items can be accelerated by a GPU. And recent libraries like Pytorch make it nearly as simple to write a GPU accelerated algorithm as a regular CPU algorithm.

    In this talk we’ll explain what the mean-shift clustering algorithm is, and why it’s important for many data science applications. We’ll first implement it in python (with numpy), and will then show how to port it to Pytorch, showing how to get a 20x performance improvement in the process.

    Upcoming Courses

    Previously (1)

    Want to teach at ga?

    Join our global community of instructors and help shape the next generation of industry leaders — while moving your own career forward with proven subject matter expertise, leadership experience, and public speaking skills. Teach online or on campus, full-time
or part-time.

    Learn More