pytorch/pytorch : 2.8.0-cuda12.9-cudnn9-runtime

PyTorch is a deep learning framework that puts Python first.

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