Building Torchvision for CUDA compute architecture 3.0 ...
github.com › pytorch › visionJan 30, 2020 · Before reading this issue, had built pytorch from source with TORCH_CUDA_ARCH_LIST=3.0 and this occurs when building torchvision with it, but editing the 'cpp_extension.py' file per above to add '3.0' for consumer grade kepler cards clears it. Would be nice if said file picked up 'TORCH_CUDA_ARCH_LIST' when building torchvision. Author