Skip to content

Commit b4e940b

Browse files
committed
Address fixes
1 parent 053e974 commit b4e940b

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

intermediate_source/torchrec_interactive_tutorial.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""
2-
**Open Source Installation** (For Reference)
3-
--------------------------------------------
2+
Introduction to TorchRec
3+
==================================
44
55
Requirements: - python >= 3.9
66
@@ -13,18 +13,20 @@
1313
1414
"""
1515

16-
# Install stable versions for best reliability
16+
###############################################
17+
# If you do not have the following components in your environment, please install them by running:
18+
19+
#
20+
# .. code-block:: sh
21+
22+
# !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U
23+
# !pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121
24+
# !pip3 install torchmetrics==1.0.3
25+
# !pip3 install torchrec --index-url https://download.pytorch.org/whl/cu121
1726

18-
!pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U
19-
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121
20-
!pip3 install torchmetrics==1.0.3
21-
!pip3 install torchrec --index-url https://download.pytorch.org/whl/cu121
2227

2328

2429
######################################################################
25-
# Intro to TorchRec
26-
# =================
27-
#
2830
# Embeddings
2931
# ~~~~~~~~~~
3032
#
@@ -1078,4 +1080,4 @@ def forward(self, kjt: KeyedJaggedTensor):
10781080
# `dlrm <https://github.com/facebookresearch/dlrm/tree/main/torchrec_dlrm/>`__
10791081
# example, which includes multinode training on the criteo terabyte
10801082
# dataset, using Meta’s `DLRM <https://arxiv.org/abs/1906.00091>`__.
1081-
#
1083+
#

0 commit comments

Comments
 (0)