Skip to content

Commit 440b1c6

Browse files
committed
update cuda version in readme
1 parent c2d89a7 commit 440b1c6

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

README.MD

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,13 +65,13 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
6565

6666
3. Install FBGEMM.
6767
```
68-
CUDA 12.4
68+
CUDA 12.8
6969
70-
pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu124
70+
pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu128
7171
72-
CUDA 12.1
72+
CUDA 12.6
7373
74-
pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu121
74+
pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu126
7575
7676
CUDA 11.8
7777
@@ -92,7 +92,7 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
9292
python setup.py install develop
9393
```
9494

95-
5. Test the installation (use torchx-nightly for 3.11; for 3.12, torchx currently doesn't work).
95+
5. Test the installation.
9696
```
9797
GPU mode
9898

0 commit comments

Comments
 (0)