CUDA Error encountered in <function CompiledLib.bn_forward_cuda at 0x7fef222d0400>

irfanICMLL / structure_knowledge_distillation

The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.

BSD 2-Clause "Simplified" License

694 stars 104 forks source link

CUDA Error encountered in <function CompiledLib.bn_forward_cuda at 0x7fef222d0400> #41

Closed zhLawliet closed 3 years ago

zhLawliet commented 3 years ago

Tesla V100 PCIe 32 GB python3.6.10 torch.version 0.4.1 CUDA Version 9.0.176 cudnn 7.1.3

i think that may be "The CUDA kernels are compiled for sm_50, sm_52 and sm_61 by default, so they won't be compatible with V00 card",how can i change the build.sh

zhLawliet commented 3 years ago

-gencode=arch=compute_70,code=sm_70

http://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/