cuda9.0编译caffe报错nvcc fatal : Unsupported gpu architecture 'compute_70'

Tesla V100 cuda9.0 caffe编译的时候报上述错误,修改方法:

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

将前面两行注释掉,再次编译。

 

posted @ 2018-02-28 10:27  rainsoul  Views(11279)  Comments(0Edit  收藏  举报