apex编译错误解决方案

最新推荐文章于 2025-10-13 12:10:15 发布

原创最新推荐文章于 2025-10-13 12:10:15 发布 · 4.9k 阅读

1 ·

CC 4.0 BY-SA版权

无专栏收录该内容

6 篇文章

订阅专栏

这里写自定义目录标题

apex编译错误解决方案

csrc/mlp.cpp:123:3: note: in expansion of macro ‘AT_DISPATCH_FLOATING_TYPES_AND_HALF’
   AT_DISPATCH_FLOATING_TYPES_AND_HALF(inputs[0].type(), "mlp_backward", [&] {
   ^
csrc/mlp.cpp:149:39: error: expected primary-expression before ‘)’ token
         work_space.data_ptr<scalar_t>(),
                                       ^
/home/wangxu/anaconda3/envs/dense_matching_benchmark/lib/python3.7/site-packages/torch/include/ATen/Dispatch.h:12:12: note: in definition of macro ‘AT_PRIVATE_CASE_TYPE’
     return __VA_ARGS__();                          \
            ^
csrc/mlp.cpp:123:3: note: in expansion of macro ‘AT_DISPATCH_FLOATING_TYPES_AND_HALF’
   AT_DISPATCH_FLOATING_TYPES_AND_HALF(inputs[0].type(), "mlp_backward", [&] {
   ^
error: command 'gcc' failed with exit status 1

#解决方法

git clone https://github.com/NVIDIA/apex.git
cd apex
git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
python setup.py install --cuda_ext --cpp_ext