PyPI page
Home page
Author:
PyTorch Team
License:
BSD-3-Clause
Summary:
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Latest version:
2.5.1
Required dependencies:
filelock
|
fsspec
|
jinja2
|
networkx
|
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-curand-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvtx-cu12
|
setuptools
|
sympy
|
triton
|
typing-extensions
Optional dependencies:
opt-einsum
|
optree
Downloads last day:
1,209,524
Downloads last week:
7,957,100
Downloads last month:
33,329,176