pytorch/pytorch : 2.5.1-cuda12.4-cudnn9-runtime

PyTorch is a deep learning framework that puts Python first.

Latest release: about 2 months ago - 14,660,913 downloads - 125 dependencies