Closed jiapei100 closed 2 years ago
cc @jianyuh -> @jiapei100 is having trouble installing fbgemm on 7.5 CUDA_ARCH, if you have any advice
@jiapei100 Do you have full error message? I was previously able to build FBGEMM in AWS g4dn instance with T4 GPU (SM75). The FBGEMM building instruction is
python setup.py install -DTORCH_CUDA_ARCH_LIST="7.5"
in fbgemm_gpu folder in https://github.com/pytorch/FBGEMM/tree/main/fbgemm_gpu
@jiapei100 -> closing this issue as we haven't heard from you, feel free to reopen if you still need help
Failed to buid torchrec from source ...
My ENV: