File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -8,6 +8,7 @@ TorchRec has been used to accelerate advancements in recommendation systems, som
88* [ Disaggregated Multi-Tower: Topology-aware Modeling Technique for Efficient Large-Scale Recommendation] ( https://arxiv.org/abs/2403.00877 ) paper
99* [ The Algorithm ML] ( https://github.com/twitter/the-algorithm-ml ) from Twitter
1010* [ Training Recommendation Models with Databricks] ( https://docs.databricks.com/en/machine-learning/train-recommender-models.html )
11+ * [ Toward 100TB model with Embedding Offloading Paper] ( https://dl.acm.org/doi/10.1145/3640457.3688037 )
1112
1213
1314## Introduction
@@ -39,6 +40,10 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
3940
40411 . Install pytorch. See [ pytorch documentation] ( https://pytorch.org/get-started/locally/ ) .
4142 ```
43+ CUDA 12.4
44+
45+ pip install torch --index-url https://download.pytorch.org/whl/nightly/cu124
46+
4247 CUDA 12.1
4348
4449 pip install torch --index-url https://download.pytorch.org/whl/nightly/cu121
@@ -60,6 +65,10 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
6065
61663 . Install FBGEMM.
6267 ```
68+ CUDA 12.4
69+
70+ pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu124
71+
6372 CUDA 12.1
6473
6574 pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu121
You can’t perform that action at this time.
0 commit comments