torchrec run failed - Githubissues

facebookresearch / dlrm

An implementation of a deep learning recommendation model (DLRM)

MIT License

3.71k stars 825 forks source link

Open zhaozheng09 opened 1 month ago

zhaozheng09 commented 1 month ago

no failed info.

[root@zw04-data-k8s-gpu-a100-node1538 torchrec_dlrm]# python3 -c "import torch; print(torch.__version__, torch.version.cuda); "
2.4.0+cu121 12.1

[root@zw04-data-k8s-gpu-a100-node1538 torchrec_dlrm]# python3 -c "import torch; import fbgemm_gpu; print(fbgemm_gpu.__version__)"
2024.8.8+cu121

| NVIDIA-SMI 530.30.02              Driver Version: 530.30.02    CUDA Version: 12.1     |

hrwleo commented 1 month ago

已收到邮件阿里阿豆故咋一马斯！如未及时回复，请致电15868848097 QQ：812737452