首页 > 其他 > 详细

NVCC src/caffe/util/math_functions.cu

时间:2018-10-15 21:11:15      阅读:207      评论:0      收藏:0      [点我收藏+]

解决办法:由于安装的cuda版本是7.5,当前下载的caffe版本比较新,需要修改里面的makefile文件,屏蔽下面的代码,cuda<8.0

 

 



In the Makefile.example, try commenting out the *_60 and *_61 lines (for compatibility with CUDA < 8.0).

CUDA_ARCH := -gencode arch=compute_20,code=sm_20         -gencode arch=compute_20,code=sm_21         -gencode arch=compute_30,code=sm_30         -gencode arch=compute_35,code=sm_35         -gencode arch=compute_50,code=sm_50         -gencode arch=compute_52,code=sm_52         #-gencode arch=compute_60,code=sm_60         #-gencode arch=compute_61,code=sm_61         #-gencode arch=compute_61,code=compute_61

注释掉

NVCC src/caffe/util/math_functions.cu

原文:https://www.cnblogs.com/huhuai/p/9794648.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!