pip install torch==1.10.0

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Source
Among top 1000 packages on PyPI.
Over 16.1M downloads in the last 90 days.

Commonly used with torch

Based on how often these packages appear together in public requirements.txt files on GitHub.

torchvision

image and video datasets and models for torch deep learning

annoy

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk.

imageio

Library for reading and writing a wide range of image, video, scientific, and volumetric data formats.

spacy

Industrial-strength Natural Language Processing (NLP) in Python

scikit-optimize

Sequential model-based optimization toolbox.

altair

Altair: A declarative statistical visualization library for Python.

tensorflow-gpu

TensorFlow is an open source machine learning framework for everyone.

sentencepiece

SentencePiece python wrapper

nbdime

Diff and merge of Jupyter Notebooks

h5py

Read and write HDF5 files from Python

joblib

Lightweight pipelining: using Python functions as pipeline jobs.

tqdm

Fast, Extensible Progress Meter

graphviz

Simple Python interface for Graphviz

zarr

An implementation of chunked, compressed, N-dimensional arrays for Python.

torchtext

Text utilities and datasets for PyTorch

kaggle

Kaggle API

bloscpack

Command line interface to and serialization format for Blosc

ipyleaflet

A Jupyter widget for dynamic Leaflet maps

hyperopt

Distributed Asynchronous Hyperparameter Optimization

Version usage of torch

Proportion of downloaded versions in the last 3 months (only versions over 1%).

1.9.0

24.50%

1.9.1

19.85%

1.10.0

12.42%

1.8.1

8.33%

1.5.0

7.91%

1.7.1

7.56%

1.5.1

4.71%

1.6.0

3.87%

1.7.0

3.18%

1.4.0

2.42%

1.8.0

1.33%

1.2.0

1.21%