码迷,mamicode.com
首页 > 其他好文 > 详细

NVCC src/caffe/util/math_functions.cu

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

标签:vcc   --   efi   gen   compute   util   arc   file   比较   

解决办法:由于安装的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

标签:vcc   --   efi   gen   compute   util   arc   file   比较   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!