-
进入caffe文件夹,在 scripts文件夹中修改build_win.cmd :
-
a.修改Cuda.cmake文件第7行: set(Caffe_known_gpu_archs “50 52 60 61 70 75”);
b. 添加99~100行内容,2080是Turing架构,算力是75.
3.然后运行build_win.cmd,在scripts\build下面会生成Caffe.sln。用vs2015打开,编译ALL_BUILD.等待编译成功。遇到的问题:
1.提示找不到_symbols.hpp,在caffe目录下找了一遍确实没找到这个文件,于是在引用处直接注释。
2.提示CMake Error at cuda_compile_1_generated_batch_norm_layer.cu.o.Release.cmake:279 (message):
Error generating file。这个挺恶心的,最终在.caffe\dependencies\libraries_v140_x64_py27_1.1.0\libraries\include\boost-1_61\boost\config\compiler\nvcc.hpp:注释掉最后三行:
if !defined(__CUDACC_VER__) || (__CUDACC_VER__ < 70500)
# define BOOST_NO_CXX11_VARIADIC_TEMPLATES
#endif
(原因:cuda7.5以后,__CUDACC_VER__被废除,应当使用__CUDACC_VER_MAJOR_和__CUDACC_VER_MINOR__等,参考https://github.com/BVLC/caffe/issues/5994、https://blog.youkuaiyun.com/qq_37817177/article/details/80604294)。
caffe原版本、编译之后的头文件和库、build_win.cmd、Cuda.cmake都上传在https://download.youkuaiyun.com/upload/11154318,想要就贡献5积分,毕竟不能拿来主义嘛~
第一次写,嘿嘿~~