Skip to content

Commit b5239c5

Browse files
committed
Address fixes 2
1 parent b4e940b commit b5239c5

File tree

1 file changed

+22
-10
lines changed

1 file changed

+22
-10
lines changed

intermediate_source/torchrec_interactive_tutorial.py

Lines changed: 22 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,33 @@
22
Introduction to TorchRec
33
==================================
44
5-
Requirements: - python >= 3.9
5+
**TorchRec** is a PyTorch library tailored for building scalable and efficient recommendation systems using embeddings.
6+
This tutorial guides you through the installation process, introduces the concept of embeddings, and highlights their importance in
7+
recommendation systems. It offers practical demonstrations on implementing embeddings with PyTorch
8+
and TorchRec, focusing on handling large embedding tables through distributed training and advanced optimizations.
9+
10+
.. grid:: 2
11+
.. grid-item-card:: :octicon:`mortar-board;1em;` What you will learn
12+
:class-card: card-prerequisites
13+
* Fundamentals of embeddings and their role in recommendation systems
14+
* How to set up TorchRec to manage and implement embeddings in PyTorch environments
15+
* Explore advanced techniques for distributing large embedding tables across multiple GPUs
16+
.. grid-item-card:: :octicon:`list-unordered;1em;` Prerequisites
17+
:class-card: card-prerequisites
18+
* PyTorch v2.5 or later with CUDA 11.8 or later
19+
* Python 3.9 or later
20+
* FBGEMM <https://github.com/pytorch/fbgemm>
621
7-
We highly recommend CUDA when using TorchRec. If using CUDA: - cuda >=
8-
11.8
9-
10-
Installing TorchRec will also install
11-
`FBGEMM <https://github.com/pytorch/fbgemm>`__, a collection of CUDA
12-
kernels and GPU enabled operations to run
1322
1423
"""
1524

1625
###############################################
17-
# If you do not have the following components in your environment, please install them by running:
18-
19-
#
26+
# Install Dependencies
27+
# ================
28+
#
29+
# Before running this tutorial in Google Colab or other environment, install the
30+
# following dependencies:
31+
#
2032
# .. code-block:: sh
2133

2234
# !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U

0 commit comments

Comments
 (0)