Ubuntu with GPU 安装caffe Makefile.config arch=compute_50,code=sm_50

设置CUDA_arch时, 是不是应该将其他的内容注释,只留下和自己电脑对应的配置

比如我的显卡是NVIDIA GTX750 , 计算能力是50,只留下-gencode arch=compute_50,code=sm_50?
可是 后面还有一行是-gencode arch=compute_50,code=compute_50
这个应该是caffe作者的失误么
 
那么 我应该这样, 对么
CUDA_ARCH := -gencode arch=compute_50,code=sm_50 \
 
 
 
最后
 这里附一下 caffe Makefile.config 源码#CUDA architecture setting: going with all of them.
# For CUDA < 6.0, comment the *_50 lines for compatibility.
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_50,code=compute_50
 
保留所有设置的话,编译的时候出现warning
nvcc warning : The 'compute_20', 'sm_20', and 'sm_21' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).
已邀请:

tianrolin

赞同来自: chg0901 xinmiao dhhcj

1.你安装的CUDA版本是8.0吧? 从CUDA 8.0开始compute capability 2.0和2.1被弃用了,所以可以将-gencode arch=compute_20,code=sm_20 和-gencode arch=compute_20,code=sm_21这两行删除。
 
2.如果将compute_30和35两行也删除的话,对于本机运行没有影响。但是如果将可执行程序发布到显卡计算能力只有3.0或3.5的机器上就不能运行了。
 
3.关于如何设置计算能力,NVIDIA官方文档在这里 http://docs.nvidia.com/cuda/cu ... ility

要回复问题请先登录注册