Closed zhaozheng09 closed 1 month ago
Hi @zhaozheng09 , could you share if you're using which torch and fbgemm versions and commands used for the installation?
Hi @zhaozheng09 , could you share if you're using which torch and fbgemm versions and commands used for the installation?
pip install torchrec==0.8.0 fbgemm==0.8.0
@zhaozheng09 thank you for the information. I assume you use torch==0.8.0?
python -c "import torch; print(torch.__version__, torch.version.cuda); "
and let me know the results? Just to confirm torch cuda version. import fbgemm_gpu
at all? Please run python -c "import torch; import fbgemm_gpu; print(fbgemm_gpu.__version__)"
I update cuda11.4->cuda12.1 it's ok .