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
8
8
* [ Disaggregated Multi-Tower: Topology-aware Modeling Technique for Efficient Large-Scale Recommendation] ( https://arxiv.org/abs/2403.00877 ) paper
9
9
* [ The Algorithm ML] ( https://github.com/twitter/the-algorithm-ml ) from Twitter
10
10
* [ 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 )
11
12
12
13
13
14
## Introduction
@@ -39,6 +40,10 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
39
40
40
41
1 . Install pytorch. See [ pytorch documentation] ( https://pytorch.org/get-started/locally/ ) .
41
42
```
43
+ CUDA 12.4
44
+
45
+ pip install torch --index-url https://download.pytorch.org/whl/nightly/cu124
46
+
42
47
CUDA 12.1
43
48
44
49
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
60
65
61
66
3 . Install FBGEMM.
62
67
```
68
+ CUDA 12.4
69
+
70
+ pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu124
71
+
63
72
CUDA 12.1
64
73
65
74
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