File tree Expand file tree Collapse file tree 1 file changed +13
-11
lines changed Expand file tree Collapse file tree 1 file changed +13
-11
lines changed Original file line number Diff line number Diff line change 1
1
"""
2
- **Open Source Installation** (For Reference)
3
- --------------------------------------------
2
+ Introduction to TorchRec
3
+ ==================================
4
4
5
5
Requirements: - python >= 3.9
6
6
13
13
14
14
"""
15
15
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
17
26
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
22
27
23
28
24
29
######################################################################
25
- # Intro to TorchRec
26
- # =================
27
- #
28
30
# Embeddings
29
31
# ~~~~~~~~~~
30
32
#
@@ -1078,4 +1080,4 @@ def forward(self, kjt: KeyedJaggedTensor):
1078
1080
# `dlrm <https://github.com/facebookresearch/dlrm/tree/main/torchrec_dlrm/>`__
1079
1081
# example, which includes multinode training on the criteo terabyte
1080
1082
# dataset, using Meta’s `DLRM <https://arxiv.org/abs/1906.00091>`__.
1081
- #
1083
+ #
You can’t perform that action at this time.
0 commit comments