Skip to content

Commit 3e58a31

Browse files
PaulZhang12facebook-github-bot
authored andcommitted
Update README with CUDA 124 (#2506)
Summary: Pull Request resolved: #2506 Reviewed By: aporialiao Differential Revision: D64765553 fbshipit-source-id: 5819468f56e3ca36c0c45a05f026548aac2e10e7
1 parent de7e041 commit 3e58a31

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

README.MD

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff 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

4041
1. 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

6166
3. 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

0 commit comments

Comments
 (0)