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
65
65
66
66
3 . Install FBGEMM.
67
67
```
68
- CUDA 12.4
68
+ CUDA 12.8
69
69
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
71
71
72
- CUDA 12.1
72
+ CUDA 12.6
73
73
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
75
75
76
76
CUDA 11.8
77
77
@@ -92,7 +92,7 @@ Check out the [Getting Started](https://pytorch.org/torchrec/setup-torchrec.html
92
92
python setup.py install develop
93
93
```
94
94
95
- 5 . Test the installation (use torchx-nightly for 3.11; for 3.12, torchx currently doesn't work) .
95
+ 5 . Test the installation.
96
96
```
97
97
GPU mode
98
98
You can’t perform that action at this time.
0 commit comments