这里写自定义目录标题
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
版权声明：本文为swfa1原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。