pytorch/pytorch : 2.7.0-cuda12.8-cudnn9-runtime

PyTorch is a deep learning framework that puts Python first.

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