环境:win10 vs2015 cuda10.1 rtx2060
任务:编译caffe
使用版本:caffe-windows,使用build_win.cmd进行编译。
问题:nvcc fatal : Unsupported gpu architecture 'compute_481'
截图:
疑问:“compute_481”的gpu架构是什么,根本查不到的啊。。。
有经验的路人帮忙解答一下啊,大恩不言谢!
--------------------------------------------------------------- 分割线 -----------------------------------------------------------------------------------------------
以上问题得以顺利解决
办法:如https://blog.csdn.net/qq_37118111/article/details/93385899所描述。
具体执行:里面涉及到build_win.cmd中的”PYTHON_VERSION“设置问题,针对2或3的python版本,将C:\Users\用户名\.caffe\dependencies\libraries_v140_x64_py27_1.1.0\libraries\include\boost-1_61\boost\config\compiler下的nvcc.hpp 或 C:\Users\用户名\.caffe\dependencies\libraries_v140_x64_py35_1.1.0\libraries\include\boost-1_61\boost\config\compiler下的nvcc.hpp下的对应部分注释。