Simulating Evolution of Multi-Particle System with GPU Acceleration in PyTorch
Published:
Author: Yeu-Guang Tung
Abstract: Simulating the evolution of a system of particles based on physical laws can be computational costly. We propose a simple way to perform GPU acceleration using PyTorch. Taking advantage of the efficient matrix operations and gradient calculations in PyTorch, we can readily accelerate the simulation using GPU. We showed that this method produces correct simulation results and runs significantly faster then when using CPU.