Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Next, you'll have to specify your GPU's architecture by editing the TORCH_CUDA_ARCH_LIST variable (find a complete list of NVIDIA GPUs and their corresponding architectures on this page).