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
文章来自:http://www.cnblogs.com/sopic/p/6680804.html
© 2021 jiaocheng.bubufx.com  联系我们
ICP备案:鲁ICP备09046678号-3