Error when Building GPU docker image for caffe: Unsupported gpu architecture 'compute_60'

issue:

Error when Building GPU docker image for caffe: Unsupported gpu architecture 'compute_60'

reason:

CUDA < 8.0

 

solution:

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
posted @ 2017-04-08 08:01  浩克匠心  阅读(561)  评论(0编辑  收藏