[go: up one dir, main page]

Sign inSign up

pytorch/pytorch

By pytorch

•Updated 17 days ago

PyTorch is a deep learning framework that puts Python first.

Image
Languages & frameworks
Machine learning & AI
Data science
1.7K

10M+

pytorch/pytorch repository overview

PyTorch is a deep learning framework that puts Python first.

It provides Tensors and Dynamic neural networks in Python with strong GPU acceleration.

http://pytorch.org⁠

Tag summary

Content type

Image

Digest

sha256:976e46afb…

Size

13.7 GB

Last updated

17 days ago

docker pull pytorch/pytorch:2.14.1-cuda12.6-cudnn9-devel