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
将前面两行注释掉,再次编译。