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

PyTorch is a deep learning framework that puts Python first.

Latest release: 3 months ago - 12,325,854 downloads - 125 dependencies