A simple command-line program to stress Nvidia GPU devices. We use PyTorch to multiply two large matrices repeatedly for some period of time.
The well-known uv utility is the best way to install this program.
uv tool install gpu-stressUse gpu-stress --help to get usage and help on options.
Use something like the following to stress the GPU for 1 minute.
gpu-stress -t 1m