jetson 编译 pytorch 出现 internal compiler error: Segmentation fault

jetson 编译 pytorch 出现：

/usr/include/c++/7/cmath: In static member function ‘static scalar_t at::native::div_floor_kernel_cuda(at::TensorIterator&)::<lambda()>::<lambda()>::<lambda(scalar_t, scalar_t)>::_FUN(scalar_t, scalar_t)’:
/usr/include/c++/7/cmath:1302:38: internal compiler error: Segmentation fault
   { return __builtin_copysignf(__x, __y); }

这是由于编译 std::copysign() 这个函数对于 Jetson 平台消耗太大。

解决方式有2种：

1. 升级 pytorch 版本到1.9，pytoch 1.9 使用 c10::cuda::compat::copysign() 函数代替了 std::copysign() 函数。

2. 按照这个提交修改代码: Workaround arm64 gcc error in std::copysign on Jetson platforms