caffe2在ubuntu 17.10 上编译成功
2018-03-04 本文已影响0人
Wendal兽
环境:
- ubuntu 17.10 amd64
- gcc 5/7
- cuda 8.0
- cuda 6.0
- python 2.7/3.5/3.6
cmake ../ -DCUDA_HOST_COMPILER="/usr/bin/gcc-5" -DCMAKE_CXX_CFLAGS="-D_GLIBCXX_USE_C99 -D_GLIBCXX_USE_C99_MATH -D__GLIBCXX__=20160904" -DCUDA_NVCC_FLAGS="-D_GLIBCXX_USE_C99" -DUSE_NATIVE_ARCH=ON
参数说明:
- CUDA_HOST_COMPILER 因为cuda 8只支持gcc5,需要设置
- 不晓得为啥gcc-7会报std::to_string不存在的之类的错误,所以启用一下相关配置
- USE_NATIVE_ARCH=ON 默认没视野aux等特性, 果断启用
