|
2 | 2 | Introduction to TorchRec
|
3 | 3 | ==================================
|
4 | 4 |
|
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> |
6 | 21 |
|
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 |
13 | 22 |
|
14 | 23 | """
|
15 | 24 |
|
16 | 25 | ###############################################
|
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 | +# |
20 | 32 | # .. code-block:: sh
|
21 | 33 |
|
22 | 34 | # !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U
|
|
0 commit comments