File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -65,13 +65,13 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
6565
66663 . 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
You can’t perform that action at this time.
0 commit comments