You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Enable Cuda 12.6 since FBGEMM offers it now (#2845)
Summary:
Pull Request resolved: #2845
I believe this should resolve cu126 errors on nightly binaries as well.
Before we filtered out CUDA version 12.6 because FBGEMM (torchrec dependency) did not support it yet, but now there is support - seen from https://download.pytorch.org/whl/nightly/cu126 with `fbgemm_gpu` available - so we should remove our filter.
Our filter was causing issue with CU126 nightly binaries because there is no torchrec version supported.
{F1976345189}
Reviewed By: aliafzal
Differential Revision: D71707060
fbshipit-source-id: 70877d6d46e749203d3f11f72973f38fa8b32f16
0 commit comments