torchrec run failed #389
Description
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 |