nvidia apex build error 编译出错 解决方法
编译 NVIDIA apex 出错解决方法
git clone https://github.com/NVIDIA/apex.git
python setup.py install --cuda_ext --cpp_ext
csrc/mlp.cpp:126:54: error: expected primary-expression before ‘>’ token
w_ptr.push_back(inputs[i + 1].data_ptr<scalar_t>());
^.
原创
2020-06-20 22:40:09 ·
3781 阅读 ·
14 评论